hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Clean + Add act_aloha_real.yaml + Add act_real.yaml

Browse Source
This commit is contained in:
Remi Cadene
2024-09-10 19:45:59 +02:00
parent a60d27b132
commit e1763aa906
4 changed files with 24 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/compute_stats.py
View File

@@ -68,7 +68,7 @@ def get_stats_einops_patterns(dataset, num_workers=0):
return stats_patterns
def compute_stats(dataset, batch_size=32, num_workers=16, max_num_samples=None):
def compute_stats(dataset, batch_size=8, num_workers=8, max_num_samples=None):
"""Compute mean/std and min/max statistics of all data keys in a LeRobotDataset."""
if max_num_samples is None:
max_num_samples = len(dataset)