2017-04-21 68 views
0

之前曾產生的工作:工作不產生/導出目錄

$ gcloud ml-engine jobs submit training testX 
    --job-dir="gs://testxxx/run1" 
    --package-path=trainer 
    --module-name=trainer.task 
    --region us-central1 
    --runtime-version=1.0 

當我看到輸出路徑的內容,我看不到「出口」 dir,只有這個:

$ gsutil ls -r $OUTPUT_PATH 
gs://testxxx/run1/: 
gs://testxxx/run1/ 
gs://testxxx/run1/packages/: 
gs://testxxx/run1/packages/fcd2eee0ae2b155ccb3b644c26cf75d6cf81b2dd068122690c9a4baf8ff8e8f5/: 
gs://testxxx/run1/packages/fcd2eee0ae2b155ccb3b644c26cf75d6cf81b2dd068122690c9a4baf8ff8e8f5/trainer-0.1.tar.gz 

我忘了任何步驟嗎?

回答

0

您提交的代碼負責導出模型。你可以在this post找到一個例子;請參考SavedModel docs。當然

輸入和模型的輸出將具體到你的模型,但爲了方便(和略有修改),這裏是從該職位代碼:

### BUILD THE PREDICTION GRAPH 
in_image = tf.placeholder(tf.uint8, shape=(None,)) 
out_classes = build_prediction_graph(in_image) 

### DEFINE SAVED MODEL SIGNATURE 
inputs = {'image_bytes': tf.saved_model.utils.build_tensor_info(in_image)} 
outputs = {'prediction': tf.saved_model.utils.build_tensor_info(out_classes)} 
signature = tf.saved_model.signature_def_utils.build_signature_def(
    inputs=inputs, 
    outputs=outputs, 
    method_name='tensorflow/serving/predict' 
) 

### SAVE OUT THE MODEL 
b = saved_model_builder.SavedModelBuilder('new_export_dir') 
b.add_meta_graph_and_variables(sess, 
           [tf.saved_model.tag_constants.SERVING], 
           signature_def_map={'serving_default': signature}) 
b.save() 
+0

謝謝你,也會在查看文檔SavedModel – Neurus