我在做 model surgery 的时候想按 Conv 0/bias 这种路径筛参数，现在 flatten dict 给的是 tuple，后面 unflatten dict 又得自己 split…
我在做 model surgery 的时候想按 Conv 0/bias 这种路径筛参数，现在 flatten dict 给的是 tuple，后面 unflatten dict 又得自己 split…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Problem Statement 我在做 model surgery 的时候想按 Conv 0/bias 这种路径筛参数，现在 flatten dict 给的是 tuple，后面 unflatten dict 又得自己 split…
The task
Problem Statement 我在做 model surgery 的时候想按 Conv_0/bias 这种路径筛参数，现在 flatten_dict 给的是 tuple，后面 unflatten_dict 又得自己 split 回去，挺绕的；能不能让这两个 traverse_util 工具直接支持用分隔符拼出来的字符串路径？另外文档里的优化器例子还在用 flax.optim，我现在项目都按 Optax 写，最好也一起换成 Optax 的写法，不然照着改参数状态时有点对不上。 Expected outcomes…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.