具身开发者社区

外观

踩坑GitHub openvla/openvla7 个月前

OpenVLA微调报错Dataset bridge_orig not found的解决与绕过 ​

OpenVLA LoRA微调时tensorflow_datasets报bridge_orig未找到，需检查数据路径配置或改用data_mix参数加载数据集。

AI 整理
  • 数据与采集
  • 具身模型

本文由 AI 根据公开内容整理，可能有误，请以原文为准：tensorflow_datasets.core.registered.DatasetNotFoundError: Dataset bridge_orig not found ERROR with LoRA finetune（GitHub openvla/openvla #255）

状态：有绕过办法

问题 ​

在OpenVLA进行LoRA微调时，尽管已按README下载并重命名数据集为bridge_orig，且能在openvla/prismatic/vla/datasets/rlds/oxe中看到注册信息，运行时仍抛出tensorflow_datasets.core.registered.DatasetNotFoundError: Dataset bridge_orig not found异常，无法识别自定义RLDS数据集。

环境 ​

OpenVLA框架，Prismatic VLA RLDS数据加载管线，TensorFlow Datasets环境，LoRA微调任务。

解决办法 ​

  1. 仔细核对代码中设置的dataset路径与data_root_dir根路径是否与实际存放bridge_orig数据的目录完全一致，修正路径配置后重试。
  2. 若路径无误仍报错，参考prismatic/vla/datasets/datasets.py中RLDSDataset实现，将加载参数由dataset改为data_mix；当传入名称不在OXE_NAMED_MIXTURES中时，代码会自动将其封装为单数据集混合配置[(self.data_mix, 1.0)]，从而绕过TFDS直接注册检查机制。

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

来源

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

  1. GitHub openvla/openvla #255：tensorflow_datasets.core.registered.DatasetNotFoundError: Dataset bridge_orig not found ERROR with LoRA finetune访问 2026-10-26