ray.rllib.algorithms.algorithm.Algorithm.import_policy_model_from_h5#

Algorithm.import_policy_model_from_h5(import_file: str, policy_id: str = 'default_policy') → None[source]#

Imports a policy’s model with given policy_id from a local h5 file.

Parameters

  • import_file – The h5 file to import from.

  • policy_id – Optional policy id to import into.

Example

>>> from ray.rllib.algorithms.ppo import PPO
>>> algo = PPO(...) 
>>> algo.import_policy_model_from_h5("/tmp/weights.h5") 
>>> for _ in range(10): 
>>>     algo.train()