{"payload":{"feedbackUrl":"https://github.com/orgs/community/discussions/53140","repo":{"id":770453014,"defaultBranch":"main","name":"tdmpc2-jax","ownerLogin":"ShaneFlandermeyer","currentUserCanPush":false,"isFork":false,"isEmpty":false,"createdAt":"2024-03-11T15:22:19.000Z","ownerAvatar":"https://avatars.githubusercontent.com/u/13738359?v=4","public":true,"private":false,"isOrgOwned":false},"refInfo":{"name":"","listCacheKey":"v0:1718197256.0","currentOid":""},"activityList":{"items":[{"before":"af6a05a3e797738ef1eeae1275369cade0a0fc6d","after":"948f86c71a2a0c50cfd4573ecd7843ca743d4753","ref":"refs/heads/develop","pushedAt":"2024-07-31T20:35:13.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Removed relative imports in envs subfolder","shortMessageHtmlLink":"Removed relative imports in envs subfolder"}},{"before":null,"after":"af6a05a3e797738ef1eeae1275369cade0a0fc6d","ref":"refs/heads/feature/cross-q","pushedAt":"2024-06-12T13:00:56.000Z","pushType":"branch_creation","commitsCount":0,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge pull request #6 from edoust/develop\n\nRemoved world model's dependency on gymnasium spaces","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2329626196\" data-permission-text=\"Title is private\" data-url=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/issues/6\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/ShaneFlandermeyer/tdmpc2-jax/pull/6/hovercard\" href=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/pull/6\">#6</a> from edoust/develop"}},{"before":"000c90c2c058f7e733a32dd5d0bcc78bd85255d0","after":"d9d95b51b04b39c45d6f16c0b907ace90871bcbf","ref":"refs/heads/main","pushedAt":"2024-06-11T15:57:28.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Squashed commit of the following:\n\ncommit af6a05a3e797738ef1eeae1275369cade0a0fc6d\nMerge: 2b6bdc0 be0eac7\nAuthor: Shane Flandermeyer <Shaneflandermeyer@gmail.com>\nDate:   Tue Jun 11 08:52:00 2024 -0700\n\n    Merge pull request #6 from edoust/develop\n\n    Removed world model's dependency on gymnasium spaces\n\ncommit be0eac79b4038e2025bbc501788b8a1945a4ba44\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Tue Jun 11 10:49:27 2024 -0500\n\n    Change default UTD ratio in config\n\ncommit 5691dc55c7e727ef3c9e1ad1c82b541d7cd8b1c2\nAuthor: edoust <edoust@users.noreply.github.com>\nDate:   Tue Jun 11 13:01:30 2024 +0200\n\n    Using env_config.num_envs instead of env.num_envs for compatibility\n\ncommit 26c42afee89c7cd8ede042ebb57db93a1207360b\nMerge: afcf3cb 2b6bdc0\nAuthor: edoust <6137595+edoust@users.noreply.github.com>\nDate:   Mon Jun 10 22:23:15 2024 +0200\n\n    Merge branch 'ShaneFlandermeyer:develop' into develop\n\ncommit 2b6bdc0e13ef186070a1f956ed192b5a41ae62fc\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 09:39:13 2024 -0500\n\n    Remove finished items from contribution list\n\ncommit afcf3cb158adfceee920b068c2ff23db610a36f7\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 09:33:40 2024 -0500\n\n    Decouple encoder and world model, formatting\n\ncommit fae1edff8408465703be33249d5c6e6e02868357\nMerge: f4d04ef 8fe0dbe\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 09:32:37 2024 -0500\n\n    Merge commit '8fe0dbe14b82e31ed5616556292093f16d6f774c' into pr/edoust/6-1\n\ncommit 8fe0dbe14b82e31ed5616556292093f16d6f774c\nMerge: 1cea473 64295be\nAuthor: Shane Flandermeyer <Shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 05:53:47 2024 -0700\n\n    Merge pull request #5 from edwhu/develop\n\n    DMC tasks\n\ncommit 64295be749873673014d9d229a1e41af7fa81c0f\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 07:52:07 2024 -0500\n\n    Update readme with comparison figures\n\ncommit 6b0af0feaae2b6c16643fcbda00bd3d2a3032583\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 10 07:39:51 2024 -0500\n\n    Add comparison images with original repo\n\ncommit 54b81fe5dfcf70c9dabffcbc539c003295a6adb3\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Jun 5 14:17:43 2024 -0500\n\n    Zero out grad nans, remove unused losses\n\ncommit dd1e69441788aa7aec0ec7d09ed45d840f0bff11\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Tue Jun 4 15:41:07 2024 -0500\n\n    Prevent tf/tensorboard GPU memory problem\n\ncommit b9401e588bc64eae45d9165021ceb0801fad54a1\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 3 14:46:33 2024 -0500\n\n    MLP LayerNorm dtype, agent.update() logging fix\n\ncommit a53bbc545b09ed61aa4f86eee26c991ba4183eca\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 3 11:25:08 2024 -0500\n\n    Use full ensemble in actor loss and value estimation (slightly better performance and matches REDQ theory)\n\ncommit 8f6ec8729bbe66c21d818f53aa604d6d3fdcc098\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 3 11:07:07 2024 -0500\n\n    More elegant world model loss\n\ncommit 26e50901cd6331c8c4ed10bb4ccf382d5d5929dc\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Jun 3 10:47:19 2024 -0500\n\n    Fix faulty termination flag in DMControl gym wrapper\n\ncommit c26c54a7d76c590319ed542a111cce324a5eb594\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun Jun 2 18:55:31 2024 -0500\n\n    WIP: Trying to match original implementation more closely\n\ncommit f4d04efb7d68d62f58e14b722f43876e1425c13e\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun Jun 2 09:06:28 2024 -0500\n\n    Update obs/action inputs to world model and fix type hints\n\ncommit 43426578acb6f165a2923f1641f741dff1bc0110\nAuthor: edoust <edoust@users.noreply.github.com>\nDate:   Sun Jun 2 14:08:24 2024 +0200\n\n    Removed world model's dependency on gymnasium spaces\n\ncommit 824de841eba24fa2488b2da5f821f35f152508d6\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 31 22:59:22 2024 -0500\n\n    Change default dtype in config to float32 and use jnp.percentile for value scale\n\ncommit 9433023b49faea77cdcf025b3c5a2a7c527739d8\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 21:58:02 2024 -0400\n\n    rename benchmark to backend\n\ncommit 4abb7ce0d93b861e493fc6133a20c585bc03e97f\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 21:27:39 2024 -0400\n\n    use jax.tree.map\n\ncommit 8e6b054d558df0804f100a14557a4e60cb39e886\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 21:27:19 2024 -0400\n\n    include replay buffer in checkpoint\n\ncommit 37d568b1c7a2897b1513045e004525e9470beca1\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 14:55:59 2024 -0400\n\n    add planning iteration heuristic\n\ncommit a9d2bcb625ccba94119d5aa74cf51ab2c2d1aa04\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 14:28:13 2024 -0400\n\n    move repeat action wrapper to envs/wrappers. it's also unused.\n\ncommit 5b7b2f14fb106c8e08151c9cfa9437757c46f495\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 14:26:33 2024 -0400\n\n    dm half cheetah running\n\ncommit 443ebce1df1faecf3ec3608add628bece0581bc7\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 13:36:51 2024 -0400\n\n    copy over tdmpc2 environment code\n\ncommit fb2304a2b03577c198ea4df97c669698a874f897\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 13:14:08 2024 -0400\n\n    use gymnasium action scale\n\ncommit 810854b32c89a4acaf4eadc8321d21b16e696715\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Fri May 31 13:02:40 2024 -0400\n\n    start log interval from initial step\n\ncommit 1cea473ee41180616c6c07b5620740c331b534cc\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 31 11:41:06 2024 -0500\n\n    Update readme\n\ncommit 88e2fd47f87207159cf38584832bdeddd5f5bd94\nMerge: c8a82b7 deac4df\nAuthor: Shane Flandermeyer <Shaneflandermeyer@gmail.com>\nDate:   Fri May 31 11:32:29 2024 -0500\n\n    Merge pull request #4 from edwhu/eh/logging\n\n    Orbax checkpointing, Tensorboard Logging\n\ncommit deac4dfd65399754af53fdb5160b7c843334a4ff\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 31 11:30:43 2024 -0500\n\n    Revert \"Fix context manager error from checkpoint object\"\n\n    This reverts commit 4445d93dd9753bd2d7a6452aa1e50705eb01ec92.\n\ncommit 4445d93dd9753bd2d7a6452aa1e50705eb01ec92\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 31 11:16:55 2024 -0500\n\n    Fix context manager error from checkpoint object\n\ncommit 534d3307d0f09ee11dd4acfc484259694a44fd96\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Thu May 30 22:07:03 2024 -0400\n\n    match paper hyperparams\n\ncommit 703f331ea9a4fd03eea8c4f48a331be0b1c9b817\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Thu May 30 22:02:08 2024 -0400\n\n    set max grad norm to 20\n\ncommit 2784e1d84f700d3ecfe4a47747fd4026147c116a\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Thu May 30 22:01:45 2024 -0400\n\n    orbax checkpointing and tensorboard logging\n\ncommit 584159c464cb886047df665ae6c75d995f3aab5e\nAuthor: Edward Hu <husedward@gmail.com>\nDate:   Thu May 30 22:01:05 2024 -0400\n\n    missing hydra dep\n\ncommit c8a82b740d35f4d883960c8d76b926e24dd76fed\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 24 09:05:27 2024 -0500\n\n    Cleanup\n\ncommit e5786a61a3590310d559d86b87ac22bd61f0a5ff\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Thu May 23 16:12:14 2024 -0500\n\n    Add warm start option for mean AND std (only using mean in driver)\n\ncommit fd738d229707b76bb5aa00405bea523f77e65fe3\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed May 22 08:08:24 2024 -0500\n\n    Change default activation to None for NormedLinear\n\ncommit ab491afbebee965c226e4c64e23b16cb204696e2\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun May 12 08:08:05 2024 -0500\n\n    Fix new problems in infinite horizon envs\n\ncommit 36e6c811e1e7e874f433408c506886b998f4ab99\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sat May 11 17:57:53 2024 -0500\n\n    Continue prediction loss no longer discounted temporally. Seems to help stabilize finite-horizon envs.\n\n    Also removed warm starting in MPC planning, as it seems to hurt performance in some environments.\n\ncommit 195183d7b4d649f42a13e0aec943406f53f7a560\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 10 21:36:44 2024 -0500\n\n    Use mean Q for actor loss scale.\n\n    Use full ensemble for value estimation during planning\n\n    Remove soft continue prediction in favor of binary\n\n    Fix warm start for finite-horizon envs\n\ncommit 2fd84868c0b00826bd93ece55f23b3c363ea957b\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri May 10 15:22:34 2024 -0500\n\n    Update readme\n\ncommit f811e870012328c871a9d49ac91cb4b7ec509f73\nMerge: ab0d966 e177680\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon May 6 17:30:27 2024 -0500\n\n    Merge commit 'e1776809f3f9bdebe5f61b6165dc5fb71da22c3c' into develop\n\ncommit ab0d9665469f417667e8287dc8102423e5536805\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon May 6 00:32:49 2024 -0500\n\n    Enabled continue predictor for example train script (needed for humanoid-v4)\n\ncommit 846437d8105344c227f9f92f6da38796cda826fb\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun May 5 22:54:47 2024 -0500\n\n    Amend\n\ncommit 91992105467f069cc56aa263875b6b034b6067ff\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun May 5 18:23:39 2024 -0500\n\n    Fix initialization keys for world model submodules\n\ncommit 41740e7ae574ed703de46a8d64d11e18f0f8c790\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Apr 22 14:56:40 2024 -0500\n\n    Added normalization/activation option for NormedLinear layer\n\ncommit 85c17b69d5fdb5ac3bffe290089f3cedd0a3f18d\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri Apr 19 08:25:36 2024 -0500\n\n    Remove action nan to num in plan()\n\ncommit a13d8589eb3f2156a93545d0388598cbf703cedf\nMerge: f851d97 29d3cd0\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Apr 17 09:36:32 2024 -0500\n\n    Merge branch 'feature/vectorized' into develop\n\ncommit 29d3cd00b540a5d1700a583625c3d58ae1ed9c62\nAuthor: ShaneFlandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Apr 17 09:35:57 2024 -0500\n\n    Added encoder optimizer option in world model\n\ncommit d07c2ed7861a1706a79d52b8ca38ac8a4b43fb96\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Mar 27 09:51:02 2024 -0500\n\n    Fix typo in config filename\n\ncommit 3e9442d7fcd8ec027f53d19ec56e7fb1362bc5a0\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Mar 27 09:18:48 2024 -0500\n\n    Using vectorized envs for main training script\n\ncommit 3b2afda432f2c17680f79b1877553efd6392a954\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Mar 27 09:13:00 2024 -0500\n\n    Added an asynchronous option in env parameters\n\ncommit 26376edc8c8f4c6ef53fe68d90b5be7d2edeb605\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Wed Mar 27 08:21:37 2024 -0500\n\n    Updated config with vectorized params\n\ncommit 152b556d9a0572e566a06d3a7dcbd6b296eac7d9\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Tue Mar 26 17:26:47 2024 -0500\n\n    Amend\n\ncommit 06459c9cc41cb2f0c8c1d3f63300df8ba65e1170\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Tue Mar 26 14:12:53 2024 -0500\n\n    Simpler sequential replay buffer and parallel env execution\n\ncommit ba38343e3ede09d7c242b8cacc683a186e2f1421\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 25 12:40:07 2024 -0500\n\n    vmap planning function for parallel execution\n\ncommit f851d9794eb3f00b420fc547168d596dd20fd7e1\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri Mar 22 11:32:17 2024 -0500\n\n    Made obs symlog optional\n\ncommit 319db3d243640c070e35489303c2a6907105e38c\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 18 14:14:23 2024 -0500\n\n    Fixed scale problem\n\ncommit 3820889c6e1508e6171e7fbe7e1b58c6ca40fc7e\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 18 11:38:25 2024 -0500\n\n    Amend\n\ncommit 0f29b1dc77a1eec9d6bbe9a569aeb64f1b0ca7d8\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 18 11:34:02 2024 -0500\n\n    Amend\n\ncommit 882dae0dc5689c022ca37672890557df40aa6b76\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 18 10:43:58 2024 -0500\n\n    Fix train script seeding on reset, add separate terminal/truncated args to update()\n\ncommit b4c91ab9e9e9e4e53a1e437dc2115c44d41f8ad4\nMerge: e6f5191 0d38516\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Mon Mar 18 09:13:48 2024 -0500\n\n    Merge commit '0d385163d79a37dfa8da089a938e29ac8f0f950e' into develop\n\ncommit e6f51911231f18a478177fbf4fe5c6217ecd5067\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sun Mar 17 08:28:31 2024 -0500\n\n    Fix typo in value loss computation\n\ncommit 03dcb388f89b7aa9b072a14283c5af78d1b5efc8\nMerge: 0aa87be e6e83b4\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sat Mar 16 14:46:19 2024 -0500\n\n    Merge branch 'feature/continue-predictor' into develop\n\ncommit e6e83b487b2223e2118e311e04ba0d3329aed8b6\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sat Mar 16 14:22:09 2024 -0500\n\n    Reduced logging frequency, slightly changed hyperparameters\n\ncommit 9d41d1849f477a7a089b90395bb41f52749c63d2\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sat Mar 16 12:40:22 2024 -0500\n\n    Verified performance in infinite horizon case and fixed continue model flag\n\ncommit 48e4453c2a9151528fdcbb8fa31f56fa7fb7959f\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Sat Mar 16 11:39:08 2024 -0500\n\n    ~Immaculate~ results in humanoid.\n\n    Only need to verify performance in infinite horizon case before merging\n\ncommit 9c45d262111e24d3e6957f8ac87843bb26517f00\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri Mar 15 15:08:48 2024 -0500\n\n    WIP: Made continue prediction optional, improved train script logging\n\ncommit 0d5083de3230ef29650882b0ea1945a4782bb3e6\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Fri Mar 15 08:30:32 2024 -0500\n\n    WIP: It seems that planning has to be the culprit here\n\ncommit 03cd3f0ebe12bcf17d7620ddb44648b03f1cdf3b\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Thu Mar 14 23:09:48 2024 -0500\n\n    Removed action scaling from TDMPC objects (replaced by wrapper)\n\ncommit 76f606198976e7a5ee5db56034a51d1205847a39\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Thu Mar 14 22:50:58 2024 -0500\n\n    WIP: Debugging humanoid env. Action scaling in planner seems like the most likely culprit\n\ncommit 0c97b21ace53046d36c94375bfb72a254c31c5eb\nAuthor: Shane Flandermeyer <shaneflandermeyer@gmail.com>\nDate:   Thu Mar 14 15:54:21 2024 -0500\n\n    Data type option in config, continue predictor in world model","shortMessageHtmlLink":"Squashed commit of the following:"}},{"before":"2b6bdc0e13ef186070a1f956ed192b5a41ae62fc","after":"af6a05a3e797738ef1eeae1275369cade0a0fc6d","ref":"refs/heads/develop","pushedAt":"2024-06-11T15:52:00.000Z","pushType":"pr_merge","commitsCount":8,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge pull request #6 from edoust/develop\n\nRemoved world model's dependency on gymnasium spaces","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2329626196\" data-permission-text=\"Title is private\" data-url=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/issues/6\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/ShaneFlandermeyer/tdmpc2-jax/pull/6/hovercard\" href=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/pull/6\">#6</a> from edoust/develop"}},{"before":"8fe0dbe14b82e31ed5616556292093f16d6f774c","after":"2b6bdc0e13ef186070a1f956ed192b5a41ae62fc","ref":"refs/heads/develop","pushedAt":"2024-06-10T14:39:16.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Remove finished items from contribution list","shortMessageHtmlLink":"Remove finished items from contribution list"}},{"before":"1cea473ee41180616c6c07b5620740c331b534cc","after":"8fe0dbe14b82e31ed5616556292093f16d6f774c","ref":"refs/heads/develop","pushedAt":"2024-06-10T12:53:47.000Z","pushType":"pr_merge","commitsCount":20,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge pull request #5 from edwhu/develop\n\nDMC tasks","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2328393318\" data-permission-text=\"Title is private\" data-url=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/issues/5\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/ShaneFlandermeyer/tdmpc2-jax/pull/5/hovercard\" href=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/pull/5\">#5</a> from edwhu/develop"}},{"before":"88e2fd47f87207159cf38584832bdeddd5f5bd94","after":"1cea473ee41180616c6c07b5620740c331b534cc","ref":"refs/heads/develop","pushedAt":"2024-05-31T16:41:10.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Update readme","shortMessageHtmlLink":"Update readme"}},{"before":"c8a82b740d35f4d883960c8d76b926e24dd76fed","after":"88e2fd47f87207159cf38584832bdeddd5f5bd94","ref":"refs/heads/develop","pushedAt":"2024-05-31T16:32:29.000Z","pushType":"pr_merge","commitsCount":7,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge pull request #4 from edwhu/eh/logging\n\nOrbax checkpointing, Tensorboard Logging","shortMessageHtmlLink":"Merge pull request <a class=\"issue-link js-issue-link\" data-error-text=\"Failed to load title\" data-id=\"2326831901\" data-permission-text=\"Title is private\" data-url=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/issues/4\" data-hovercard-type=\"pull_request\" data-hovercard-url=\"/ShaneFlandermeyer/tdmpc2-jax/pull/4/hovercard\" href=\"https://github.com/ShaneFlandermeyer/tdmpc2-jax/pull/4\">#4</a> from edwhu/eh/logging"}},{"before":"87ad3a6b41f0607a33d6948b0a5bb7f3b24a0212","after":"000c90c2c058f7e733a32dd5d0bcc78bd85255d0","ref":"refs/heads/main","pushedAt":"2024-05-30T13:47:55.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Update README.md","shortMessageHtmlLink":"Update README.md"}},{"before":"e5786a61a3590310d559d86b87ac22bd61f0a5ff","after":"c8a82b740d35f4d883960c8d76b926e24dd76fed","ref":"refs/heads/develop","pushedAt":"2024-05-24T14:05:30.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Cleanup","shortMessageHtmlLink":"Cleanup"}},{"before":"fd738d229707b76bb5aa00405bea523f77e65fe3","after":"e5786a61a3590310d559d86b87ac22bd61f0a5ff","ref":"refs/heads/develop","pushedAt":"2024-05-23T21:12:17.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Add warm start option for mean AND std (only using mean in driver)","shortMessageHtmlLink":"Add warm start option for mean AND std (only using mean in driver)"}},{"before":"ab491afbebee965c226e4c64e23b16cb204696e2","after":"fd738d229707b76bb5aa00405bea523f77e65fe3","ref":"refs/heads/develop","pushedAt":"2024-05-22T13:08:58.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Change default activation to None for NormedLinear","shortMessageHtmlLink":"Change default activation to None for NormedLinear"}},{"before":"e1776809f3f9bdebe5f61b6165dc5fb71da22c3c","after":"87ad3a6b41f0607a33d6948b0a5bb7f3b24a0212","ref":"refs/heads/main","pushedAt":"2024-05-15T14:20:01.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge commit 'ab491afbebee965c226e4c64e23b16cb204696e2'","shortMessageHtmlLink":"Merge commit 'ab491afbebee965c226e4c64e23b16cb204696e2'"}},{"before":"f811e870012328c871a9d49ac91cb4b7ec509f73","after":"ab491afbebee965c226e4c64e23b16cb204696e2","ref":"refs/heads/develop","pushedAt":"2024-05-13T13:57:46.000Z","pushType":"push","commitsCount":4,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Fix new problems in infinite horizon envs","shortMessageHtmlLink":"Fix new problems in infinite horizon envs"}},{"before":"ab0d9665469f417667e8287dc8102423e5536805","after":"f811e870012328c871a9d49ac91cb4b7ec509f73","ref":"refs/heads/develop","pushedAt":"2024-05-06T22:30:31.000Z","pushType":"push","commitsCount":3,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge commit 'e1776809f3f9bdebe5f61b6165dc5fb71da22c3c' into develop","shortMessageHtmlLink":"Merge commit 'e1776809f3f9bdebe5f61b6165dc5fb71da22c3c' into develop"}},{"before":"4260f6e0d653073b5d177d4c023ecc6d5c2abd8d","after":"e1776809f3f9bdebe5f61b6165dc5fb71da22c3c","ref":"refs/heads/main","pushedAt":"2024-05-06T22:26:13.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Vectorized environments","shortMessageHtmlLink":"Vectorized environments"}},{"before":"846437d8105344c227f9f92f6da38796cda826fb","after":"ab0d9665469f417667e8287dc8102423e5536805","ref":"refs/heads/develop","pushedAt":"2024-05-06T05:32:53.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Enabled continue predictor for example train script (needed for humanoid-v4)","shortMessageHtmlLink":"Enabled continue predictor for example train script (needed for human…"}},{"before":"91992105467f069cc56aa263875b6b034b6067ff","after":"846437d8105344c227f9f92f6da38796cda826fb","ref":"refs/heads/develop","pushedAt":"2024-05-06T03:54:52.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Amend","shortMessageHtmlLink":"Amend"}},{"before":"41740e7ae574ed703de46a8d64d11e18f0f8c790","after":"91992105467f069cc56aa263875b6b034b6067ff","ref":"refs/heads/develop","pushedAt":"2024-05-05T23:23:42.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Fix initialization keys for world model submodules","shortMessageHtmlLink":"Fix initialization keys for world model submodules"}},{"before":"85c17b69d5fdb5ac3bffe290089f3cedd0a3f18d","after":"41740e7ae574ed703de46a8d64d11e18f0f8c790","ref":"refs/heads/develop","pushedAt":"2024-04-22T19:56:42.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Added normalization/activation option for NormedLinear layer","shortMessageHtmlLink":"Added normalization/activation option for NormedLinear layer"}},{"before":"a13d8589eb3f2156a93545d0388598cbf703cedf","after":"85c17b69d5fdb5ac3bffe290089f3cedd0a3f18d","ref":"refs/heads/develop","pushedAt":"2024-04-19T13:25:40.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Remove action nan to num in plan()","shortMessageHtmlLink":"Remove action nan to num in plan()"}},{"before":"f851d9794eb3f00b420fc547168d596dd20fd7e1","after":"a13d8589eb3f2156a93545d0388598cbf703cedf","ref":"refs/heads/develop","pushedAt":"2024-04-17T14:36:36.000Z","pushType":"push","commitsCount":9,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Merge branch 'feature/vectorized' into develop","shortMessageHtmlLink":"Merge branch 'feature/vectorized' into develop"}},{"before":"29d3cd00b540a5d1700a583625c3d58ae1ed9c62","after":null,"ref":"refs/heads/feature/vectorized","pushedAt":"2024-04-17T14:36:34.000Z","pushType":"branch_deletion","commitsCount":0,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"}},{"before":"d07c2ed7861a1706a79d52b8ca38ac8a4b43fb96","after":"29d3cd00b540a5d1700a583625c3d58ae1ed9c62","ref":"refs/heads/feature/vectorized","pushedAt":"2024-04-17T14:36:21.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Added encoder optimizer option in world model","shortMessageHtmlLink":"Added encoder optimizer option in world model"}},{"before":"3e9442d7fcd8ec027f53d19ec56e7fb1362bc5a0","after":"d07c2ed7861a1706a79d52b8ca38ac8a4b43fb96","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-27T14:51:06.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Fix typo in config filename","shortMessageHtmlLink":"Fix typo in config filename"}},{"before":"3b2afda432f2c17680f79b1877553efd6392a954","after":"3e9442d7fcd8ec027f53d19ec56e7fb1362bc5a0","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-27T14:18:53.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Using vectorized envs for main training script","shortMessageHtmlLink":"Using vectorized envs for main training script"}},{"before":"26376edc8c8f4c6ef53fe68d90b5be7d2edeb605","after":"3b2afda432f2c17680f79b1877553efd6392a954","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-27T14:13:04.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Added an asynchronous option in env parameters","shortMessageHtmlLink":"Added an asynchronous option in env parameters"}},{"before":"152b556d9a0572e566a06d3a7dcbd6b296eac7d9","after":"26376edc8c8f4c6ef53fe68d90b5be7d2edeb605","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-27T13:21:41.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Updated config with vectorized params","shortMessageHtmlLink":"Updated config with vectorized params"}},{"before":"06459c9cc41cb2f0c8c1d3f63300df8ba65e1170","after":"152b556d9a0572e566a06d3a7dcbd6b296eac7d9","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-26T22:27:04.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Amend","shortMessageHtmlLink":"Amend"}},{"before":"ba38343e3ede09d7c242b8cacc683a186e2f1421","after":"06459c9cc41cb2f0c8c1d3f63300df8ba65e1170","ref":"refs/heads/feature/vectorized","pushedAt":"2024-03-26T19:12:57.000Z","pushType":"push","commitsCount":1,"pusher":{"login":"ShaneFlandermeyer","name":"Shane Flandermeyer","path":"/ShaneFlandermeyer","primaryAvatarUrl":"https://avatars.githubusercontent.com/u/13738359?s=80&v=4"},"commit":{"message":"Simpler sequential replay buffer and parallel env execution","shortMessageHtmlLink":"Simpler sequential replay buffer and parallel env execution"}}],"hasNextPage":true,"hasPreviousPage":false,"activityType":"all","actor":null,"timePeriod":"all","sort":"DESC","perPage":30,"cursor":"djE6ks8AAAAEjnhaPQA","startCursor":null,"endCursor":null}},"title":"Activity · ShaneFlandermeyer/tdmpc2-jax"}