具身开发者社区

外观

踩坑GitHub Physical-Intelligence/openpi11 天前

RoboArena训练配置缺失weight_loader导致模型随机初始化 ​

RoboArena五个配置项未指定PaliGemma权重加载器，导致训练从随机初始化开始；修复方案为显式设置weight_loader。

AI 整理
  • 仿真与训练
  • 具身模型

本文由 AI 根据公开内容整理，可能有误，请以原文为准：RoboArena config weight initialization missing（GitHub Physical-Intelligence/openpi #856）

状态：已解决

问题 ​

在openpi仓库中，src/openpi/training/misc/roboarena_config.py内的五个TrainConfig（paligemma_binning_droid等）均未配置weight_loader字段，默认回退至NoOpWeightLoader，致使模型训练从随机权重初始化而非预期的PaliGemma预训练权重启动，与代码注释描述不符。

环境 ​

Physical-Intelligence/openpi仓库，涉及src/openpi/training/misc/roboarena_config.py及src/openpi/training/config.py:480默认配置逻辑。

解决办法 ​

  1. 修改src/openpi/training/misc/roboarena_config.py中全部五个RoboArena基线配置项，显式添加weight_loader=weight_loaders.PaliGemmaWeightLoader()参数。
  2. 确保导入路径正确引用src/openpi/training/weight_loaders.py中的PaliGemmaWeightLoader类。
  3. 添加回归测试用例，断言每个相关配置对象的weight_loader属性均为PaliGemmaWeightLoader实例，防止后续回退至NoOpWeightLoader。

讨论原作者：@albertwilcox 等，版权归原作者所有。

来源

本页事实对应的公开页面，以原文为准。

  1. GitHub Physical-Intelligence/openpi #856：RoboArena config weight initialization missing访问 2026-10-26