ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-02 11:51:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

General fixes to abide by the new config in learner_server, actor_server, gym_manipulator

Browse Source
This commit is contained in:
Michel Aractingi
2025-05-27 15:49:33 +02:00
committed by AdilZouitine
parent df96e5b3b2
commit 1edfbf792a
8 changed files with 49 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/server/learner_server.py
View File

@@ -91,6 +91,7 @@ from torch import nn
from torch.multiprocessing import Queue
from torch.optim.optimizer import Optimizer
from lerobot.common.cameras import so100_follower_end_effector # noqa: F401
from lerobot.common.constants import (
CHECKPOINTS_DIR,
LAST_CHECKPOINT_LINK,
@@ -101,6 +102,8 @@ from lerobot.common.datasets.factory import make_dataset
from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
from lerobot.common.policies.factory import make_policy
from lerobot.common.policies.sac.modeling_sac import SACPolicy
from lerobot.common.robots import so100_follower_end_effector # noqa: F401
from lerobot.common.teleoperators import gamepad, so100_leader # noqa: F401
from lerobot.common.utils.random_utils import set_seed
from lerobot.common.utils.train_utils import (
get_step_checkpoint_dir,