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

small fix

This commit is contained in:
Cadene 2024-04-17 20:48:13 +00:00
parent 7c11568279
commit 6b137f0b80
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/train.py
View File

@ -5,7 +5,7 @@ from pathlib import Path
import hydra import hydra
import torch import torch
from datasets import Dataset, concatenate_datasets from datasets import Dataset, concatenate_datasets
from datasets.utils.logging import disable_progress_bar, enable_progress_bars from datasets.utils import disable_progress_bars, enable_progress_bars
from lerobot.common.datasets.factory import make_dataset from lerobot.common.datasets.factory import make_dataset
from lerobot.common.datasets.utils import cycle from lerobot.common.datasets.utils import cycle
@ -177,7 +177,7 @@ def add_episodes_inplace(
example["episode_data_index_to"] += start_index example["episode_data_index_to"] += start_index
return example return example
disable_progress_bar() # map has a tqdm progress bar disable_progress_bars() # map has a tqdm progress bar
hf_dataset = hf_dataset.map(shift_indices) hf_dataset = hf_dataset.map(shift_indices)
enable_progress_bars() enable_progress_bars()