tangger/issacdataengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rich annotations & update open-pi fsdp explanations

Browse Source
This commit is contained in:
Leon998
2026-03-18 13:59:52 +08:00
parent 814f3c3526
commit 4934c4794e
11 changed files with 349 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
policy/openpi-InternData-A1/src/openpi/training/config.py
View File

@@ -1814,7 +1814,6 @@ _CONFIGS = [
pytorch_weight_path="",
num_train_steps=2_000_000,
num_workers=12,
fsdp_devices=8,
batch_size=512,
save_interval=5000,
lr_schedule=_optimizer.WarmupConstantSchedule(),
@@ -1844,7 +1843,6 @@ _CONFIGS = [
pytorch_weight_path="",
num_train_steps=30_000,
num_workers=32,
fsdp_devices=8,
batch_size=128,
save_interval=5000,
),
@@ -1872,7 +1870,6 @@ _CONFIGS = [
pytorch_weight_path="",
num_train_steps=30_000,
num_workers=32,
fsdp_devices=8,
batch_size=128,
save_interval=5000,
),
@@ -1901,4 +1898,4 @@ def check_lerobot_repo(repo_dir: str):
if os.path.isdir(os.path.join(repo_dir, "data")) and os.path.isdir(os.path.join(repo_dir, "meta")) and os.path.isdir(os.path.join(repo_dir, "videos")):
return True
else:
return False
return False