Question

我正在测试一个主要为它提供服务的简单模型。我在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'```

Answer 1

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

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

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

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

1 个答案: