admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 13:09:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add resume

Browse Source
This commit is contained in:
glannuzel
2025-08-07 11:29:15 +02:00
parent 33b6a4ddcd
commit 20f5f45ac2
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/robots/reachy2/data_acquisition_server.py
+10
View File
@@ -211,6 +211,16 @@ class DataAcquisitionServicer():
obs_features = hw_to_dataset_features(self.robot.observation_features, "observation") obs_features = hw_to_dataset_features(self.robot.observation_features, "observation")
dataset_features = {**action_features, **obs_features} dataset_features = {**action_features, **obs_features}
print(f"Dataset name: {request.dataset_name}")
if request.resume:
self.dataset = LeRobotDataset(
request.dataset_name,
root="/home/demo/.cache/huggingface/lerobot/" + request.dataset_name,
)
if hasattr(self.robot, "cameras") and len(self.robot.cameras) > 0:
self.dataset.start_image_writer()
else:
# Create the dataset # Create the dataset
self.dataset = LeRobotDataset.create( self.dataset = LeRobotDataset.create(
repo_id=request.dataset_name, repo_id=request.dataset_name,