Commit 81db0b2
feature/polish(pu): add atari/dmc multitask and balance pipeline in ScaleZero paper and fix MuZero/UniZero longrun performance (#451)
* feature(pu): add unizero/muzero multitask pipeline and net plasticity related metrics
* fix(pu): fix some adaptation bug
* feature(pu): add unizero multitask balance pipeline for atari and dmc
* fix(pu): fix some adaptation bug
* feature(pu): add vit encoder for unizero
* polish(pu): polish moe layer in transformer
* feature(pu): add eval norm mean/medium for atari
* fix(pu): fix atari norm mean/median, fix collect in balance pipeline
* polish(pu): polish config
* fix(pu): fix dmc multitask to be compatiable with timestep (which is used in rope)
* polish(pu): polish config
* fix(pu): fix task_id bug in balance pipeline, and polish benchmark_name option
* fix(pu): fix benchmark_name option
* polish(pu): fix norm score computation, adapt config to aliyun
* polish(pu): polish unizero_mt balance pipeline use CurriculumController and fix solved gpu batch-size bug
* tmp
* tmp
* tmp
* test(pu): add vit moe test
* polish(pu): add adapter_scales to tb
* feature(pu): add atari uz balance config
* polish(pu): add stable_adaptor_scale
* tmp
* sync code
* polish(pu): use freeze_non_lora_parameters in transformer, not use LearnableScale in balance pipeline
* feature(pu): add vit-encoder lora in balance pipeline
* polish(pu): fix reanalyze index bug, fix global_solved bug, add apply_curriculum_to_encoder option
* polish(pu): add collect/eval_num_simulations option
* polish(pu): polish comments and style in entry of scalezero
* polish(pu): polish comments and style of ctree/tree_search/buffer/common.py
* polish(pu): polish comments and style of files in lzero.model
* polish(pu): polish comments and style of files in lzero.model.unizero_world_models
* polish(pu): polish comments and style of unizero_world_models
* polish(pu): polish comments and style of files in policy/
* polish(pu): polish comments and style of files in worker
* polish(pu): polish comments and style of files in configs
* fix(pu): fix some merge typo
* fix(pu): fix ln norm_type, fix kv_cache rewrite bug, add value_priority, fix _reset_collect/eval, add adaptive policy entropy control
* fix(pu): fix unizero_mt
* polish(pu): add LN in head, polish init_weight, polish adamw
weight-decay
* fix(pu): fix configure_optimizer_unizero in unizero_mt
* feature(pu): add encoder-clip, label smooth, analyze_latent_representation option in unizero.py
* feature(pu): add encoder-clip, label smooth option in unizero_multitask.py
* fix(pu): fix tb log when gpu_num<task_num, fix total_loss += bug, polish
alpha_loss
* polish(pu):polish config
* fix(pu): fix encoder-clip bug and num_channel/res bug
* polish(pu): polish scale_factor in DPS
* tmp
* feature(pu): add some analysis metrics in tensorboard for unizero and unizero-mt
* polish(pu): abstract a KVCacheManager for world model
* tmp
* polish(pu): polish unizero obs_loss to cos_sim loss
* tmp
* polish(pu): polish minotor-log and adapt to ale/xxx-v5 style game
* feature(pu): add decode_loss for unizero atari
* test(pu): test unizero-mt
* fix(pu): fix Deep Copy Before Storag bug when Use KVCacheManager
* sync code
* feature(pu): add iter_policy_evaluation demo in grid-world
* polish(pu): polish atari uz config
* polish(pu): polish policy logits stability
* sync code
* polish(pu): polish policy logits stability
* fix(pu): fix exp_name and task_id bug in dmc pipeline, fix some configs
* feature(pu): add head-clip manager
* fix(pu): fix head-clip log
* tmp
* polish(pu): polish comments and code styles
* polish(pu): polish comments and code styles in entry/mcts/model
* polish(pu): polish comments and code styles in policy/config
* polish(pu): polish comments and code styles in config
* polish(pu): polish comments and code styles in atari env
* fix(pu): fix comments of worker in ddp mode, fix device bug in evaluator for unizero_multitask pipeline
* fix(pu): fix unizero_multitask ddp barrier bug
* fix(pu): add policy_logits_clip_method option
* fix(pu): add policy_logits_clip_method option
* polish(pu): polish comments, docstring, readme
* polish(pu): polish atari unizero configs and default configs in unizero.py and unizero.py
* polish(pu): update to macos-15
* fix(pu): fix gymnasium[atari] version
* fix(pu): fix import bug
* polish(pu): polish comments, docstring, some little redundancy
* polish(pu): optimize import orders
* refactor(pu): move some reusable common var. and safe_eval() method to lzero/entry/utils.py
* fix(pu): fix Optional import bug
* fix(pu): fix prediction network
* fix(pu): add brew install swig in test.yml
* fix(pu): fix import bug in test
* fix(pu): fix type lint bug
* fix(pu): fix import bug in test
* fix(pu): fix import bug in test
* fix(pu): fix test
* fix(pu): fix some args bug
* polish(pu): add some comments and little polish
* fix(pu): fix 2 tests
* fix(pu): fix not_enough_data ddp bug
* fix(pu): fix final_norm_option and predict_latent_loss_type default config bug
---------
Co-authored-by: puyuan <puyuan1996@qq.com>
Co-authored-by: zjowowen <zjowowen@outlook.com>
Co-authored-by: jasper <1157507000@qq.com>
Co-authored-by: wangshulun <wangshulun@vivi-x.ai>1 parent 8ec0169 commit 81db0b2
File tree
96 files changed
+21064
-3437
lines changed- .github/workflows
- lzero
- entry
- mcts
- buffer
- tests
- tree_search
- model
- tests
- unizero_world_models
- policy
- tests/config
- worker
- zoo
- atari
- config
- envs
- dmc2gym
- config
- envs
- jericho/configs
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
96 files changed
+21064
-3437
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
53 | 53 | | |
54 | 54 | | |
55 | 55 | | |
56 | | - | |
| 56 | + | |
57 | 57 | | |
58 | 58 | | |
59 | 59 | | |
| |||
73 | 73 | | |
74 | 74 | | |
75 | 75 | | |
76 | | - | |
| 76 | + | |
77 | 77 | | |
78 | | - | |
| 78 | + | |
79 | 79 | | |
80 | | - | |
| 80 | + | |
81 | 81 | | |
82 | 82 | | |
83 | 83 | | |
| |||
167 | 167 | | |
168 | 168 | | |
169 | 169 | | |
170 | | - | |
| 170 | + | |
171 | 171 | | |
172 | 172 | | |
173 | | - | |
| 173 | + | |
174 | 174 | | |
175 | | - | |
| 175 | + | |
176 | 176 | | |
177 | 177 | | |
178 | | - | |
| 178 | + | |
179 | 179 | | |
180 | | - | |
| 180 | + | |
181 | 181 | | |
182 | 182 | | |
183 | | - | |
| 183 | + | |
184 | 184 | | |
185 | | - | |
| 185 | + | |
186 | 186 | | |
187 | 187 | | |
188 | | - | |
| 188 | + | |
189 | 189 | | |
190 | 190 | | |
191 | 191 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
56 | 56 | | |
57 | 57 | | |
58 | 58 | | |
59 | | - | |
| 59 | + | |
60 | 60 | | |
61 | 61 | | |
62 | 62 | | |
| |||
76 | 76 | | |
77 | 77 | | |
78 | 78 | | |
79 | | - | |
| 79 | + | |
80 | 80 | | |
81 | | - | |
| 81 | + | |
82 | 82 | | |
83 | | - | |
| 83 | + | |
84 | 84 | | |
85 | 85 | | |
86 | 86 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
20 | 20 | | |
21 | 21 | | |
22 | 22 | | |
23 | | - | |
| 23 | + | |
24 | 24 | | |
25 | 25 | | |
26 | 26 | | |
| |||
61 | 61 | | |
62 | 62 | | |
63 | 63 | | |
64 | | - | |
| 64 | + | |
65 | 65 | | |
66 | 66 | | |
67 | 67 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1453 | 1453 | | |
1454 | 1454 | | |
1455 | 1455 | | |
1456 | | - | |
| 1456 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
| 91 | + | |
| 92 | + | |
| 93 | + | |
| 94 | + | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
| 98 | + | |
| 99 | + | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
| 112 | + | |
| 113 | + | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
| 117 | + | |
| 118 | + | |
| 119 | + | |
| 120 | + | |
| 121 | + | |
| 122 | + | |
| 123 | + | |
| 124 | + | |
| 125 | + | |
| 126 | + | |
| 127 | + | |
| 128 | + | |
| 129 | + | |
| 130 | + | |
| 131 | + | |
| 132 | + | |
| 133 | + | |
| 134 | + | |
| 135 | + | |
| 136 | + | |
| 137 | + | |
| 138 | + | |
| 139 | + | |
| 140 | + | |
| 141 | + | |
| 142 | + | |
| 143 | + | |
| 144 | + | |
| 145 | + | |
| 146 | + | |
| 147 | + | |
| 148 | + | |
| 149 | + | |
| 150 | + | |
| 151 | + | |
| 152 | + | |
| 153 | + | |
| 154 | + | |
| 155 | + | |
| 156 | + | |
0 commit comments