feat(tj): update unizero ppo implementation#473
Open
tAnGjIa520 wants to merge 15 commits into
Open
Conversation
…files and update configurations
- Replace manual GAE computation with ding.rl_utils.gae_data and gae - Keep original implementation as _batch_compute_gae_for_pool_bak for backup - Add test script to verify GAE computation correctness - Fix lunarlander_env.py to handle both int and numpy array actions - Add lunarlander_disc_unizero_ppo_config.py for PPO training
…d game demos - Refine train_unizero, game buffers, world_model, unizero policy and collector - Add policy utils (GAE-related); remove test_gae_computation.py - Add LunarLander online PPO config; update Atari/LunarLander PPO configs - Add GAME_README, GAE plan, lunarlander flow doc, game scripts and requirements Made-with: Cursor
Ensure returns are derived from raw value+advantage before adv normalization to keep target values correct. Made-with: Cursor
…O value loss - Restructure CLAUDE.md with LightZero framework overview - Add quick start commands, testing instructions, and architecture guide - Document PPO modifications with detailed data flow - Include development patterns and debugging tips - Fix PPO value loss: remove gamma^t discount from value loss computation - Keep gamma^t discount only for world model losses (obs/rewards) - Align with standard PPO value loss formulation Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
- Add Claude Code directories to .gitignore (.codex, codex_home/, .omc/, .claude/, scripts/, docs/) - Update CLAUDE.md with environment setup and PPO implementation details - Refine split training mode in train_unizero.py for better World Model/PPO separation - Enhance buffer tracking with latest_push_count and new_data_ratio in game_buffer.py - Improve PPO batch extraction and padding logic in game_buffer_unizero.py - Add loss_type parameter support in world_model.py for flexible training modes - Update PPO configurations with split training options and improved hyperparameters Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
c10fa1a to
48b72b2
Compare
Remove docs/ directory from version control as specified in .gitignore. The directory will remain locally but won't be tracked by git. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Main Changes
functions in muzero_collector
still correctly uses gamma^t for temporal discounting)
主要改动