ExportManagerBase
ExportManager public API interface.
ExportManagerBase
-
class orbax.export.export_manager_base.ExportManagerBase[source][source]
Define the base class API which manages the exporting of a model.
-
abstract save(model_path, **kwargs)[source][source]
Saves the JAX model to a Savemodel.
-
abstract load(model_path, **kwargs)[source][source]
Load the model from a Savemodel path.