Skip to content

feature(pu): add atari/dmc multitask and balance pipeline in ScaleZero paper #417

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 39 commits into
base: main
Choose a base branch
from
Open

feature(pu): add atari/dmc multitask and balance pipeline in ScaleZero paper #417

wants to merge 39 commits into from
+16,406 −2,475

Conversation

puyuan1996
Copy link
Collaborator

@puyuan1996 puyuan1996 commented Sep 18, 2025 •
edited
Loading

puyuan and others added 29 commits April 25, 2025 11:26
feature(pu): add unizero/muzero multitask pipeline and net plasticity…
da4cdba 
… related metrics
fix(pu): fix some adaptation bug
a6eed25
feature(pu): add unizero multitask balance pipeline for atari and dmc
67a0e9a
fix(pu): fix some adaptation bug
f083096
feature(pu): add vit encoder for unizero
37eb118
polish(pu): polish moe layer in transformer
f32d63e
feature(pu): add eval norm mean/medium for atari
c0aa747
fix(pu): fix atari norm mean/median, fix collect in balance pipeline
8b3cff6
polish(pu): polish config
f2c158b
fix(pu): fix dmc multitask to be compatiable with timestep (which is …
20b42f7 
…used in rope)
polish(pu): polish config
39ee55e
fix(pu): fix task_id bug in balance pipeline, and polish benchmark_na…
e85c449 
…me option
fix(pu): fix benchmark_name option
c16d564
polish(pu): fix norm score computation, adapt config to aliyun
474b81c
polish(pu): polish unizero_mt balance pipeline use CurriculumControll…
50e367e 
…er and fix solved gpu batch-size bug
tmp
9171c3e
Merge branch 'dev-multitask-balance-clean' of https://github.com/open…
bc5003a 
…dilab/LightZero into dev-multitask-balance-clean
tmp
158e4a0
tmp
d66b986
test(pu): add vit moe test
0d5ede0
polish(pu): add adapter_scales to tb
ca6ddb6
feature(pu): add atari uz balance config
7dd6c04
polish(pu): add stable_adaptor_scale
c8e7cb8
tmp
0313335
sync code
ef170fd
@zjowowen
polish(pu): use freeze_non_lora_parameters in transformer, not use Le…
bbec353 
…arnableScale in balance pipeline
@zjowowen
feature(pu): add vit-encoder lora in balance pipeline
20648d5
polish(pu): fix reanalyze index bug, fix global_solved bug, add apply…
db6032a 
…_curriculum_to_encoder option
polish(pu): add collect/eval_num_simulations option
f63b544
@puyuan1996 puyuan1996 added the research Research work in progress label Sep 18, 2025
@puyuan1996 puyuan1996 added enhancement New feature or request config New or improved configuration labels Sep 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
config New or improved configuration enhancement New feature or request research Research work in progress
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@puyuan1996 @zjowowen @tAnGjIa520