ray.rllib.models.torch.torch_modelv2.TorchModelV2.import_from_h5

TorchModelV2.import_from_h5(h5_file: str) → None

Imports weights from an h5 file.

Parameters

h5_file – The h5 file name to import weights from.

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()