tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add MultiLerobotDataset for training with multiple LeRobotDatasets (#229)

Browse Source
This commit is contained in:
Alexander Soare
2024-05-30 16:12:21 +01:00
committed by GitHub
parent 265b0ec44d
commit 111cd58f8a
8 changed files with 352 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/configs/default.yaml
View File

@@ -23,6 +23,10 @@ use_amp: false
# `seed` is used for training (eg: model initialization, dataset shuffling)
# AND for the evaluation environments.
seed: ???
# You may provide a list of datasets here. `train.py` creates them all and concatenates them. Note: only data
# keys common between the datasets are kept. Each dataset gets and additional transform that inserts the
# "dataset_index" into the returned item. The index mapping is made according to the order in which the
# datsets are provided.
dataset_repo_id: lerobot/pusht
training: