openpi
openpi copied to clipboard
Physical-Intelligence
Created a new `MixtureDataset` Dataset type that supports weighted sampling from a list of datasets. Inspired by the [octo](https://github.com/octo-models/octo/tree/main) codebase that uses TensorFlow's [sample_from_datasets](https://www.tensorflow.org/api_docs/python/tf/data/Dataset#sample_from_datasets) to create oxe mixtures. Since torch...
### Problem: When following the README's Docker build process and running training, the container crashes with: ImportError: libGL.so.1: cannot open shared object file This occurs during OpenCV initialization, preventing training...
The "raw_dataset = tfds.load(raw_dataset_name, data_dir=data_dir, split="train")" code in "convert_libero_data_to_lerobot.py" reports an error. How to solve it
Hi, I am trying to get the Pi0 Fine Tuned Lora model to run on my computer so I can train it further on my own dataset. I believe it...
OS: 22.04 GPU: 5090 docker compose error docker compose -f scripts/docker/compose.yml up --build Compose can now delegate builds to bake for better performance. To do so, set COMPOSE_BAKE=true. [+] Building...
I have been fine-tuning the provided pi0-base model on my dataset using LeRobot. After training for 100,000 steps, I found that the model performs well on tasks that appeared in...
Thank you so much for sharing the excellent codebase! I’m currently working on a new robot, the Agilex Cobot Magic, which is an another version of the Mobile Aloha. I’ve...
Hi @kpertsch, thanks for the great work! I wanted to ask, would you perhaps mind sharing LR schedule, batch size etc for pi0 and pi0-FAST DROID finetunes? I'm trying to...
Hello! I have been trying to use `openpi` as a package using the instructions provided. Everytime iI try to import ``` openpi.policies.policy_config openpi.training.config ``` I get the following problems: ```...
