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=None, runner_cls=None, policy_obs_dims_getter=None, sim2sim_preflight=None, runner_loader=None, guard_algo_name=None, log=<built-in function print>)[source]
Create a playback session and load the selected policy checkpoint.
- Parameters:
playback_cfg (PlaybackConfig)
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]] | None)
sim2sim_preflight (Callable[[str | None], Any] | None)
runner_loader (Callable[[Any, str], None] | None)
guard_algo_name (str | None)
log (Callable[[str], None])
- Return type:
tuple[RslRlPlaybackSession, str, str | None]