TensorFlow Estimator clear_deivces у экспортеров? - PullRequest
Новая
       8

TensorFlow Estimator clear_deivces у экспортеров?

0 голосов
/ 21 ноября 2018

В вопросе SO загрузка SavedModel с другим расположением устройства рассматривается, как использовать API Estimator для сохранения модели и загрузки ее на другое устройство.

Согласно нескольким вопросам, после сохранения модели с ручным размещением device это невозможно.Однако, согласно @ Yiding :

ваш вопрос 1 может работать, только если вы очистите все устройства при экспорте модели в исходном коде тензорного потока с флагом clear_devices=True.

Если мы посмотрим на документы TensorFlow для BestExporter

__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 не вариант.

Так, как я могу обойти это?

...