Skip to content
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

feature(pu): unizero efficiency optimization and ddp configs #274

Open
wants to merge 21 commits into
base: main
Choose a base branch
from
Open

feature(pu): unizero efficiency optimization and ddp configs #274

wants to merge 21 commits into from

Conversation

puyuan1996
Copy link
Collaborator

No description provided.

PaParaZz1 and others added 21 commits August 19, 2024 01:17
@PaParaZz1
polish(pu): polish quantize_state_hash and deepcopy
95886bd
@PaParaZz1
fix(pu): fix np.array dtype bug in buffer
0e49a30
@PaParaZz1
polish(pu): use 0 deepcopy in kv_cache operation in collect/eval phas…
00147f4 
…e of unizero
@PaParaZz1
polish(pu): use custom deepcopy for kv_cache
b40c71b
@PaParaZz1
polish(pu): use value_array rather than value_list in compute_target_…
2cc81be 
…value
@PaParaZz1
polish(pu): optimize compute_target_policy_non_re
bc5332f
@PaParaZz1
polish(pu): optimize kv_caching update()
a6c6a8e
@PaParaZz1
polish(pu): kv_cache_dict no to_cpu
b5dcdcc
@PaParaZz1
polish(pu): optimize custom kv_cache copy
5b0cbd4
@PaParaZz1
polish(pu): kv_cache_dict no to_cpu
0035829
@PaParaZz1
feature(pu): add unizero ddp config
043727b
@dyyoungg
fix(pu): fix unizero ddp
d568008
@dyyoungg
sync code
d349137
@PaParaZz1
polish(pu): use de kv_cacheepcopy only in recur_infer load
3a344aa
@dyyoungg
Merge branch 'dev-efficiency' of https://github.com/opendilab/LightZero…
40053f7 
… into dev-efficiency
@dyyoungg
sync code
61a1139
@dyyoungg
sync code
bb38a10
@dyyoungg
Merge branch 'dev-efficiency' of https://github.com/opendilab/LightZero…
b813be7 
… into dev-efficiency
@jiayilee65
polish(pu): use share_polol for kv_cache in recurrent_inference and u…
715d17e 
…se _copy rather than clone
@jiayilee65
polish(pu): all kv_cache copy use predefined share_pool
39d6bbe
@dyyoungg
polish(pu): unuse decoder_net and lpips in ddp config
f18be2a
@puyuan1996 puyuan1996 added enhancement New feature or request efficiency optimization Efficiency optimization (time, memory and so on) labels Sep 5, 2024
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
efficiency optimization Efficiency optimization (time, memory and so on) enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
4 participants
@puyuan1996 @PaParaZz1 @dyyoungg @jiayilee65