Tensorflow多重导出策略

时间:2017-07-04 14:48:52

标签: tensorflow google-cloud-ml google-cloud-ml-engine

我正在使用Experiment class培训模型,虽然文档似乎建议您可以有多个导出策略:

  

export_strategies:可以导出ExportStrategys,或者单个,或者无。

当我加入两个时,我在使用ml引擎进行训练时出错:

 AssertionError: Export directory already exists. Please specify a different export directory

使用make_export_strategy function时,没有指定导出目录的选项。

我是以错误的方式接近这个吗?最终,我希望人们能够使用CSV和JSON输入向同一模型发出预测请求。

 tf.contrib.learn.Experiment(
    estimator=estimator,
    train_input_fn=train_input,
    eval_input_fn=eval_input,
    eval_metrics=eval_metrics,
    train_steps=train_steps,
    eval_steps=eval_steps,
    eval_delay_secs=eval_delay_secs,
    min_eval_frequency=min_eval_frequency,        
    export_strategies=[
      saved_model_export_utils.make_export_strategy(
        serving_input_fn=csv_serving_input_fn,
        exports_to_keep=1),
      saved_model_export_utils.make_export_strategy(
        serving_input_fn=json_serving_input_fn,
        exports_to_keep=1)
    ]
 )

0 个答案:

没有答案