hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Cleanup policies imports

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 11:08:40 +01:00
parent 6909b62a21
commit 80387285bb
15 changed files with 53 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/train.py
View File

@@ -29,8 +29,7 @@ from lerobot.common.datasets.sampler import EpisodeAwareSampler
from lerobot.common.datasets.utils import cycle
from lerobot.common.envs.factory import make_env
from lerobot.common.optim.factory import make_optimizer_and_scheduler
from lerobot.common.policies.factory import make_policy
from lerobot.common.policies.pretrained import PreTrainedPolicy
from lerobot.common.policies import PreTrainedPolicy, make_policy
from lerobot.common.policies.utils import get_device_from_parameters
from lerobot.common.utils.logging_utils import AverageMeter, MetricsTracker
from lerobot.common.utils.random_utils import set_seed