ValueError: Tensor не является элементом этого графика, когда размещает модель в Sagemaker с Gunicorn, Flask и Keras. - PullRequest
Новая
       7

ValueError: Tensor не является элементом этого графика, когда размещает модель в Sagemaker с Gunicorn, Flask и Keras.

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

Я создал предиктор временных рядов с Keras и докеризовал модель с помощью Flash и Gunicorn в соответствии с документами AWS. Я загружаю сериализованную модель с этим кодом. 

@classmethod
def get_model(cls):
    if cls.model == None:
        cls.model = load_model('/opt/ml/bitcoin_model.h5')
    return cls.model

Затем я использовал метод прогнозирования для получения результатов, докеризованный контейнер отлично работает в локальной среде, но когда я пытаюсь разместить модель в sagemaker, она выдает эту ошибку. 

ValueError: Tensor Tensor("dense_1/BiasAdd:0", shape=(?, 1), dtype=float32) is not an element of this graph.

Так, как я могу решить эту проблему?

1 Ответ

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

Проблема была решена путем вызова метода _make_predict_function () на этапе загрузки модели. 

@classmethod
def get_model(cls):
    if cls.model == None:
        cls.model = load_model('/opt/ml/bitcoin_model.h5')
        cls.model._make_predict_function()
    return cls.model

Ссылка на ошибку: https://github.com/keras-team/keras/issues/6462

...