代码之家 › 专栏 › 技术社区 › Nirojan Selvanathan

ValueError:当使用Gunicorn、Flask和Keras在Sagemaker中托管模型时,张量不是此图的元素

aws-sagemaker keras amazon-web-services

Nirojan Selvanathan · 技术社区 · 7 年前

@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 回复 | 直到 7 年前

Nirojan Selvanathan 7 年前

通过在模型加载阶段调用\u make \u predict \u 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

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

4 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

4 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

4 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

4 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

4 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

4 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

4 年前

user836026 · 在pytorch中复制权重时语法无效

4 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

4 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

8 年前