Marco Abbatangelo
Marco Abbatangelo

Reputation: 87

Vertex AI Endpoints - Failed to create endpoint

I'm trying to deploy a custom model to an endpoint with Vertex AI. I run the custom training and the model was correctly created in my bucket using Tensorflow 2 export_saved_model for estimators. In this bucket there is the saved_model.pb file with the folder variables. However, when I try to create an endpoint selecting the path to the saved model, the following error occurs:

Failed to create endpoint "endpoint_name" due to error: APPLICATION_ERROR; google.cloud.ml.v1/ModelService.CreateVersion;Field: version.deployment_uri Error: Deployment directory gs://different_bucket/artifacts/ is expected to contain exactly one of: [saved_model.pb, saved_model.pbtxt].

It seems it is searching the .pb file in a bucket that is not the one I set. Suggestions?

Upvotes: 1

Views: 1486

Answers (1)

Marco Abbatangelo
Marco Abbatangelo

Reputation: 87

I solved the problem changing the folder to pass during the endpoint creation. I have the model in the following path:

gs://my_bucket/my_folder/model/saved_model.pb

Instead of pointing to gs://my_bucket/my_folder/model/ I used gs://my_bucket/my_folder and it worked.

Upvotes: 2

Related Questions