Language Language English 简体中文

unilab.visualization.interactive_playback.create_rsl_rl_playback_session

unilab.visualization.interactive_playback.create_rsl_rl_playback_session(*, playback_cfg, env_factory, algo_config, root_dir, device, checkpoint_resolver, checkpoint_input_dim_reader, entrypoint_log_root, wrapper_cls, runner_cls, policy_obs_dims_getter, train_cfg_normalizer, log=<built-in function print>)

Create a playback session and load the selected policy checkpoint.

Parameters:

• playback_cfg (RslRlPlaybackConfig)

• env_factory (Callable[[int], Any])

• algo_config (dict[str, Any])

• root_dir (str | Path)

• device (str | None)

• checkpoint_resolver (Callable[[str, str, str | None, str, str | None], str | None])

• checkpoint_input_dim_reader (Callable[[str], int | None])

• entrypoint_log_root (Callable[..., Path])

• wrapper_cls (Any)

• runner_cls (Any)

• policy_obs_dims_getter (Callable[[Any], tuple[int, int]])

• train_cfg_normalizer (Callable[[dict[str, Any]], dict[str, Any]])

• log (Callable[[str], None])

Return type:

tuple[RslRlPlaybackSession, str, str | None]