TensorFlow Estimator clear_deivces在出口商中?

时间:2018-11-21 10:09:05

标签: python tensorflow export tensorflow-estimator

S.O。问题load SavedModel with different device placement探索了如何使用Estimator API保存模型并将其加载到其他设备上。

到目前为止,根据几个问题,一旦通过手动放置device保存了模型,这是不可行的。但是,根据@Yiding

  

只有当您在原始张量流代码中导出带有标志clear_devices=True的模型时清除所有设备时,问题1才能起作用。

如果我们查看BestExporter的TensorFlow文档

__init__(
    name='best_exporter',
    serving_input_receiver_fn=None,
    event_file_pattern='eval/*.tfevents.*',
    compare_fn=_loss_smaller,
    assets_extra=None,
    as_text=False,
    exports_to_keep=5
)

很明显clear_devices不是一个选择。

那我该如何解决呢?

0 个答案:

没有答案