ray.rllib.algorithms.algorithm.Algorithm.export_model#

Algorithm.export_model(export_formats: Union[List[str], str], export_dir: Optional[str] = None)#

Exports model based on export_formats.

Subclasses should override _export_model() to actually export model to local directory.

Parameters

  • export_formats – Format or list of (str) formats that should be exported.

  • export_dir – Optional dir to place the exported model. Defaults to self.logdir.

Returns

A dict that maps ExportFormats to successfully exported models.