ray.tune.Trainable.export_model#

Trainable.export_model(export_formats: List[str] | str, export_dir: str | None = None)[source]#

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.