ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Model loader is responsible for loading shared model types.

construct_fn A callable which creates the model instance. The callable should take no args as input (typically a closure is used to capture necessary parameters).
tags Optional model tags (e.g. 'serve' for serving or 'eval' for EvalSavedModel).



View source

Returns loaded model.

model_load_time_callback Optional callback to track load time.

construct_fn Instance of member_descriptor
tags Instance of member_descriptor