ray.rllib.algorithms.algorithm.Algorithm.export_model

Algorithm.export_model(export_formats: List[str] | str, export_dir: str | None = 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.