def _model_deploy(args, cell):
parts = args['name'].split('.')
if len(parts) == 2:
model_name, version_name = parts[0], parts[1]
model_exists = False
try:
# If describe() works, the model already exists.
datalab_ml.Models(project_id=args['project']).get_model_details(model_name)
model_exists = True
except:
pass
if not model_exists:
datalab_ml.Models(project_id=args['project']).create(model_name)
versions = datalab_ml.ModelVersions(model_name, project_id=args['project'])
runtime_version = args['runtime_version']
if not runtime_version:
runtime_version = tf.__version__
versions.deploy(version_name, args['path'], runtime_version=runtime_version)
else:
raise ValueError('Name must be like "model.version".')
评论列表
文章目录