Skip to content

Dataset Management

Dataset Management is used to inspect and organize collected data.

Its purpose is to turn collected demonstrations into reliable training material before training begins. The Collection page records demonstrations; the Dataset Management page determines whether those demonstrations are suitable for training.

The left side displays the local dataset list. Multiple datasets with exactly the same task description can be selected and merged.

Dataset merging combines data collected in separate batches for the same task, hardware configuration, and camera layout. Before merging, confirm that the task description, object placement rules, camera layout, robotic arm model, and sampling configuration are consistent. Otherwise, the merged dataset may have more samples but greater internal variation, which can reduce training quality.

  • Merge only datasets with the same task objective. Do not mix “pick up the red block” and “pick up the blue block”.
  • Merge only datasets with the same number of cameras and the same viewing angles.
  • After merging, inspect the total episode count, sample frames, joint curves, and status again.
  • Before uploading to Hugging Face, confirm that the dataset does not contain irrelevant, failed, or privacy-sensitive content.

Dataset Management screenshot 1

Dataset Management screenshot 2

Training quality depends heavily on data quality. If the dataset contains failed actions, dropped frames, inconsistent task descriptions, or large changes in camera viewpoint, the model may learn incorrect behavior.

After entering Dataset Management, do not only check whether the dataset exists. Check whether it is trainable. Trainable means the task objective is consistent, every episode is complete, video and joint data are synchronized, episode duration is reasonable, and there are no obvious corrupted frames or abnormal data.

Dataset Management screenshot 3

After selecting a dataset, you can view:

AreaContent
TaskTask description
DurationTotal duration
EpisodesNumber of episodes
StatusData quality / status
Joint state curvesJoint state curves
Video / sample frame previewVideo or sample frame preview
Episode listFrame count, duration, and preview for each episode
OperationDescription
Edit taskModify the task description
Refresh previewRegenerate or refresh previews
Upload to Hugging FaceUpload the dataset to Hugging Face
TrainUse this dataset for training
Delete episodesDelete selected episodes
MergeMerge datasets with the same task description
Replay episodeReplay a demonstration episode in the dataset

Uploading to Hugging Face requires an access token. The page provides token input, account recognition, and repository configuration.

Dataset Management screenshot 4

Check ItemWhat to Look ForWhy It Matters
Task descriptionWhether it matches the training objectiveInconsistent descriptions make one dataset express multiple goals and reduce training stability
Number of episodesWhether there are enough valid demonstrationsWith too few samples, the model may memorize individual trajectories and generalize poorly
Episode durationWhether it is much shorter than configured or contains long idle periodsAbnormal duration may indicate early interruption, dropped frames, or unclear task boundaries
Video / sample framesWhether objects, gripper, and key actions are clearly visibleVision models rely on images to judge state. Occlusion and blur affect learning
Joint curvesWhether curves are continuous, with no jumps, breaks, or abnormal shakingAbnormal curves may indicate connection, calibration, or recording issues
Status promptsWhether there are repairable or unusable statusesDetermines whether repair, re-recording, or deletion is needed

Deleting episodes is not about reducing data. It is about removing samples that may mislead the model. Failed demonstrations, obvious collisions, failed grasps, interrupted actions, and severe camera occlusion can all cause the model to learn unstable behavior. Keeping a smaller set of high-quality samples is usually more reliable than keeping a large but inconsistent dataset.

After checking, cleaning, and merging the dataset, the user should have a dataset with a clear task objective, stable quality, and training readiness. Starting training at this point makes the training process more controllable and makes it easier for the model to reproduce the intended task during execution.