diff --git a/lerobot/scripts/server/learner_server.py b/lerobot/scripts/server/learner_server.py index 0a1b0a77..98d2dbd8 100644 --- a/lerobot/scripts/server/learner_server.py +++ b/lerobot/scripts/server/learner_server.py @@ -285,7 +285,7 @@ def add_actor_information_and_train( # ds_meta=cfg.dataset, env_cfg=cfg.env, ) - + assert isinstance(policy, nn.Module) policy.train()