multi-node openpi commit

2026-03-17 23:05:23 +08:00
parent 28833f0c0f
commit 7411e0e004
156 changed files with 33951 additions and 1 deletions
--- a/policy/openpi-InternData-A1/scripts/training_scripts/single_node_multi_gpu.sh
+++ b/policy/openpi-InternData-A1/scripts/training_scripts/single_node_multi_gpu.sh
@@ -0,0 +1,13 @@
+set -ex
+
+export IMAGEIO_FFMPEG_EXE=ffmpeg
+export OMP_NUM_THREADS=128
+
+export PYTHONPATH=YOUR_PATH/openpi/src:YOUR_PATH/openpi/packages/openpi-client/src:YOUR_PATH/openpi/third_party/lerobot:${PYTHONPATH}
+conda activate pi0
+
+cd YOUR_PATH/openpi
+ulimit -n 1000000
+config_name=$1
+XLA_PYTHON_CLIENT_MEM_FRACTION=0.9  python scripts/train.py ${config_name}  \
+  --exp-name=${config_name}