trinity.manager.state_manager module

State manager.

class trinity.manager.state_manager.StateManager(path: str, trainer_name: str | None = None, explorer_name: str | None = None, config: Config | None = None, check_config: bool = False)

Bases: object

A Manager class for managing the running state of Explorer and Trainer.

__init__(path: str, trainer_name: str | None = None, explorer_name: str | None = None, config: Config | None = None, check_config: bool = False)

save_explorer(current_task_index: int, current_step: int) → None

load_explorer() → dict

save_trainer(current_exp_index: int, current_step: int) → None

load_trainer() → dict

save_stage(current_stage: int) → None

load_stage() → dict