2017-04-21 58 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