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

fix

This commit is contained in:
Ville Kuosmanen 2025-04-13 12:04:39 +02:00
parent 9b9e164cf4
commit 33f6b63a7c
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lerobot/scripts/train.py
View File

@ -24,13 +24,8 @@ from termcolor import colored
from torch.amp import GradScaler
from torch.optim import Optimizer
<<<<<<< HEAD
from lerobot.common.datasets.factory import make_dataset
=======
from lerobot.common.datasets.factory import make_dataset, resolve_delta_timestamps
from lerobot.common.datasets.lerobot_dataset import LeRobotDataset, MultiLeRobotDataset
from lerobot.common.datasets.online_buffer import OnlineBuffer, compute_sampler_weights
>>>>>>> 1c9d53c1 (fix: support multi repo datasets for training)
from lerobot.common.datasets.lerobot_dataset import MultiLeRobotDataset
from lerobot.common.datasets.sampler import EpisodeAwareSampler
from lerobot.common.datasets.utils import cycle
from lerobot.common.envs.factory import make_env