Language Language English 简体中文

unilab.utils.sim2sim.policy_load_dim_guard

unilab.utils.sim2sim.policy_load_dim_guard(*, env_obs_dim=None, env_action_dim=None, algo_name=None)

Re-raise a tensor shape mismatch during checkpoint load as a sim2sim diagnostic.

Non-matching errors propagate unchanged, so a valid load is never blocked.

Parameters:

• env_obs_dim (int | None)

• env_action_dim (int | None)

• algo_name (str | None)

Return type:

Iterator[None]