pantheonrl.algos.bc.reconstruct_policy

reconstruct_policy(policy_path, device='auto')[source]

Reconstruct a saved policy. Args:

policy_path: path where .save_policy() has been run. device: device on which to load the policy.

Returns:

policy: policy with reloaded weights.

Parameters:
  • policy_path (str) –

  • device (device | str) –

Return type:

BasePolicy