search

tensorflow / cloud

The TensorFlow Cloud repository provides APIs that will allow to easily go from debugging and training your Keras and TensorFlow code in a local environment to distributed training in the cloud.
https://github.com/tensorflow/cloud
Apache License 2.0
372 stars 85 forks source link

Trained model should be deployable/downloadable from UI. #245

Open rosbo opened 3 years ago

rosbo commented 3 years ago

When using TensorFlow Cloud, it would be great if we could deploy or download the trained model using the AI Platform UI: 8HHws5MkeedV4u5

The 2 buttons in the screenshot above are currently greyed out.

rosbo commented 3 years ago

I think TensorFlow Cloud should set the jobDir when calling the the ml.v1.projects.jobs#create api to enable at least the "download model" feature. I am not entirely sure whether this will enable the "deploy model" feature: https://github.com/tensorflow/cloud/blob/master/src/python/tensorflow_cloud/core/deploy.py#L96

ML API reference: https://cloud.google.com/ai-platform/training/docs/reference/rest/v1/projects.jobs#Job