wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

revision

This commit is contained in:
Alexander Soare 2024-04-17 09:02:35 +01:00
parent 1331d3b4e4
commit c9454333d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -70,7 +70,7 @@ def make_dataset(
transform=Prod(in_keys=clsfunc.image_keys, prod=1 / 255.0), transform=Prod(in_keys=clsfunc.image_keys, prod=1 / 255.0),
) )
stats = compute_stats(stats_dataset) stats = compute_stats(stats_dataset)
os.makedirs(precomputed_stats_path.parent, exist_ok=True) precomputed_stats_path.parent.mkdir(parents=True, exist_ok=True)
torch.save(stats, precomputed_stats_path) torch.save(stats, precomputed_stats_path)
else: else:
stats = torch.load(stats_path) stats = torch.load(stats_path)