def export_neuropod(model_path,
output_path='neuropod',
model_name='neuropod',
**kwargs):
"""Exports a model to Neuropod
# Inputs
:param model_path: (str) filepath to pre-trained model.
:param output_path: (str, default: `'neuropod'`) directory to store the
neuropod model.
:param model_name: (str, default: `'neuropod'`) save neuropod under this
name.
# Return
:returns: (`None`)
"""
logger.info('Model path: {}'.format(model_path))
logger.info('Output path: {}'.format(output_path))
logger.info('\n')
utils_export_neuropod(model_path, output_path, model_name)
logger.info('Saved to: {0}'.format(output_path))
def export_neuropod(model_path,
output_path="neuropod",
model_name="neuropod",
**kwargs):
"""Exports a model to Neuropod.
# Inputs
:param model_path: (str) filepath to pre-trained model.
:param output_path: (str, default: `'neuropod'`) directory to store the
neuropod model.
:param model_name: (str, default: `'neuropod'`) save neuropod under this
name.
# Return
:returns: (`None`)
"""
logger.info(f"Model path: {model_path}")
logger.info(f"Output path: {output_path}")
logger.info("\n")
model = LudwigModel.load(model_path)
os.makedirs(output_path, exist_ok=True)
utils_export_neuropod(model, output_path, model_name)
logger.info(f"Saved to: {output_path}")
def export_neuropod(
model_path,
output_path='neuropod',
model_name='neuropod',
**kwargs
):
"""Exports a model to Neuropod
:param model_path: Is the model from which the tensors will be collected
:param output_path: Output directory
:param debug: To step through the stack traces and find possible errors
:returns: None
"""
logger.info('Model path: {}'.format(model_path))
logger.info('Output path: {}'.format(output_path))
logger.info('\n')
utils_export_neuropod(model_path, output_path, model_name)
logger.info('Saved to: {0}'.format(output_path))