mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 10:16:09 +00:00
remove white lines
This commit is contained in:
@@ -331,6 +331,7 @@ def load_state_dict_with_missing_key_handling(
|
|||||||
# Check for missing keys
|
# Check for missing keys
|
||||||
missing_keys = load_result.missing_keys
|
missing_keys = load_result.missing_keys
|
||||||
unexpected_keys = load_result.unexpected_keys
|
unexpected_keys = load_result.unexpected_keys
|
||||||
|
|
||||||
# Filter out whitelisted missing keys
|
# Filter out whitelisted missing keys
|
||||||
policy_type_lower = policy_type.lower()
|
policy_type_lower = policy_type.lower()
|
||||||
whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, [])
|
whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, [])
|
||||||
|
|||||||
@@ -175,6 +175,7 @@ def rollout(
|
|||||||
with torch.inference_mode():
|
with torch.inference_mode():
|
||||||
action = policy.select_action(observation)
|
action = policy.select_action(observation)
|
||||||
action = postprocessor(action)
|
action = postprocessor(action)
|
||||||
|
|
||||||
action_transition = {"action": action}
|
action_transition = {"action": action}
|
||||||
action_transition = env_postprocessor(action_transition)
|
action_transition = env_postprocessor(action_transition)
|
||||||
action = action_transition["action"]
|
action = action_transition["action"]
|
||||||
@@ -510,6 +511,7 @@ def eval_main(cfg: EvalPipelineConfig):
|
|||||||
envs = make_env(cfg.env, n_envs=cfg.eval.batch_size, use_async_envs=cfg.eval.use_async_envs)
|
envs = make_env(cfg.env, n_envs=cfg.eval.batch_size, use_async_envs=cfg.eval.use_async_envs)
|
||||||
|
|
||||||
logging.info("Making policy.")
|
logging.info("Making policy.")
|
||||||
|
|
||||||
policy = make_policy(
|
policy = make_policy(
|
||||||
cfg=cfg.policy,
|
cfg=cfg.policy,
|
||||||
env_cfg=cfg.env,
|
env_cfg=cfg.env,
|
||||||
|
|||||||
@@ -200,12 +200,12 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
|
|||||||
|
|
||||||
if is_main_process:
|
if is_main_process:
|
||||||
logging.info("Creating policy")
|
logging.info("Creating policy")
|
||||||
|
|
||||||
policy = make_policy(
|
policy = make_policy(
|
||||||
cfg=cfg.policy,
|
cfg=cfg.policy,
|
||||||
ds_meta=dataset.meta,
|
ds_meta=dataset.meta,
|
||||||
rename_map=cfg.rename_map,
|
rename_map=cfg.rename_map,
|
||||||
)
|
)
|
||||||
|
|
||||||
# Wait for all processes to finish policy creation before continuing
|
# Wait for all processes to finish policy creation before continuing
|
||||||
accelerator.wait_for_everyone()
|
accelerator.wait_for_everyone()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user