export_model

Trainable.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 (List[str] | str) – Format or list of (str) formats that should be exported.

• export_dir (str | None) – Optional dir to place the exported model. Defaults to self.logdir.

Returns:

A dict that maps ExportFormats to successfully exported models.