时间戳

首页 > 解决方案 > 尝试保存张量流模型时出错

tensorflow - 尝试保存张量流模型时出错

问题描述

我正在测试一个简单的模型，主要是为它服务。我在colab环境中。执行代码

import numpy as np
from tensorflow import keras
model=tf.keras.Sequential([keras.layers.Dense(units=1,input_shape=[1])])
model.compile(optimizer='sgd', loss='mean_squared_error')
xs=np.array([-1.0,0.0,1.0,2.0,3.0,4.0],dtype=float)
ys=np.array([-3.0,-1.0,1.0,3.0,5.0,7.0],dtype=float)
model.fit(xs,ys,epochs=500,verbose=2)

tf.saved_model.simple_save(
    keras.backend.get_session(),
    export_path,
    inputs={'input_image': model.input},
    outputs={t.name:t for t in model.outputs})

我收到以下我不知道如何修复的错误：

<ipython-input-19-634675006b49> in <module>()
----> 1 tf.saved_model.simple_save(
      2     keras.backend.get_session(),
      3     export_path,
      4     inputs={'input_image': model.input},
      5     outputs={t.name:t for t in model.outputs})

AttributeError: module 'tensorflow._api.v2.saved_model' has no attribute 'simple_save'```

标签： tensorflowgoogle-colaboratorytensorflow-serving

解决方案

simple_save在 tensorflow v2 ( LINK ) 中已弃用。

尝试model.save(saving/path)改用

关于 model.save 的完整文档：https ://www.tensorflow.org/tutorials/keras/save_and_load#save_the_entire_model

推荐阅读