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
不是一个选择。
那我该如何解决呢?