rich annotations & update open-pi fsdp explanations

2026-03-18 13:59:52 +08:00
parent 814f3c3526
commit 4934c4794e
11 changed files with 349 additions and 32 deletions
--- a/policy/openpi-InternData-A1/scripts/training_scripts/multi_node.sh
+++ b/policy/openpi-InternData-A1/scripts/training_scripts/multi_node.sh
@@ -202,8 +202,6 @@ python scripts/train_jax_multinode.py \
    pretrain-interndata-a1 \
  --exp-name=pretrain-interndata-a1 \
  --num_workers=12 \
-  --fsdp_devices=8 \
  --batch_size=512 \
  --num_train_steps=2000000 \
  --save_interval=5000
-