SerializedTorchPolicy¶
-
class
maze.core.agent.serialized_torch_policy.
SerializedTorchPolicy
(model: Union[omegaconf.DictConfig, Dict], state_dict_file: str, spaces_dict_file: str, device: str)¶ Structured policy used for rollouts of trained models.
Will build the models based on the model composer and spaces config and set the state of individual policies according to the state dict dump.
Policies are set to eval mode by default.
- Parameters
model – Model composer configuration
state_dict_file – Path to dumped state dictionaries of the trained policies
spaces_dict_file – Path to dumped spaces configuration (action and observation spaces of the env the policy was trained on, used for model initialization)
-
seed
(seed: int)¶ (overrides
TorchPolicy
)Set torch manual seed