set num workers to zero in compute stats, temporary fix to hanging code when iterating over dataloader

This commit is contained in:
Michel Aractingi
2025-01-20 11:02:37 +00:00
parent acc433d25d
commit 6cc2cc896a
3 changed files with 2 additions and 4 deletions

View File

@@ -160,9 +160,8 @@ def to_lerobotdataset_with_save_episode(raw_dir: Path, repo_id: str, push_to_hub
for key in other_keys:
if "language_instruction" in key:
continue
# Some openx dataset have multiple language commands
# episode_data[key] = episode[key].numpy()[0].decode("utf-8")
episode_data[key] = episode[key].numpy()[0].decode("utf-8")
else:
if key == "is_last":
episode_data["next.done"] = tf_to_torch(episode[key])