openpi

openpi copied to clipboard

jaxlib.xla_extension.XlaRuntimeError: UNKNOWN: CUDNN_STATUS_EXECUTION_FAILED

2

When I use multi-GPU training, the following error occurs：“Traceback (most recent call last): File "/liury/src/openpi/scripts/train.py", line 304, in main(_config.cli()) File "/liury/src/openpi/scripts/train.py", line 285, in main train_state, info = ptrain_step(train_rng, train_state,...

liu-ry

Is it necessary to finetune all data in libero?

2

I noticed that pi0 performs well on Libero Spatial, Libero Object, Libero Goal, and Libero 10 datasets. However, the model I tried to pretrain on libero 90 does not work...

dstx123

State/Action representation in LIBERO dataset (EEF vs Joint)

13

Hello, thanks for your great work first! I recently downloaded the LIBERO dataset, and I noticed something confusing regarding the representation of state and action in the feature files. -...

lyfeng001

supports gradient accumulation and update README

1

yaolug

Can not run inference.ipynb in the example

8

Cell In[5], line 7 2 config = dataclasses.replace(config, batch_size=2) 4 # Load a single batch of data. This is the same data that will be used during training. 5 #...

dengpingsong

When I ran train.py in OpenPI, I got the following error:

2

Environment: Ubuntu 24.04, RTX5090, CUDA 12.6.3 When I run XLA_PYTHON_CLIENT_MEM_FRACTION=0.9 uv run scripts/train.py pi0_fast_libero --exp-name=my_experiment --overwrite I get the following error: XLA_PYTHON_CLIENT_MEM_FRACTION=0.9 uv run scripts/train.py pi0_fast_libero --exp-name=my_experiment --overwrite warning: The...

satokeeen

question about computation of the norm stats

4

hi, I see that executing `uv run scripts/compute_norm_stats.py` is a must when fine-tuning on libero dataset. however, when i fine-tune on droid dataset, and if i skip the compute norm...

yanan1116

Client failed to connect: websockets.exceptions.InvalidMessage: did not receive a valid HTTP response

3

I'm trying to deploy the OpenPI inference server by following the official instructions. The server seems to start correctly, but when I run the client, it fails with the following...

TH-666

python 3.13 install openpi will meet problem , bazel cannot find. But use python 3.11 can install smoothly.

LukeLIN-web

Fine-tune PI0 base model on Droid dataset failed when computing stats

2

Hi, I'm trying to fine-tune PI0 base model on Droid dataset. I noticed there is only a PI0-Fast-Finetune version of TrainConfig, so I defined a PI0-Finetune version TrainConfig as follows:...

AsDeadAsADodo