保存_模型.修剪（在TF2.0中）

Question

我正在尝试修剪生成的SavedModel的节点特斯拉斯. 修剪脚本如下：

svmod = tf.saved_model.load(fn) #version 1
#svmod = tfk.experimental.load_from_saved_model(fn) #version 2
feeds = ['foo:0']
fetches = ['bar:0']
svmod2 = svmod.prune(feeds=feeds, fetches=fetches)
tf.saved_model.save(svmod2, '/tmp/saved_model/') #version 1
#tfk.experimental.export_saved_model(svmod2, '/tmp/saved_model/') #version 2

如果我使用版本1，修剪可以工作，但在保存时会给出ValueError: Expected a Trackable object for export。在版本2中，没有prune（）方法。在

如何修剪TF2.0 Keras SavedModel？在

Answer 1

由于您可以在版本1中成功地修剪，所以我建议您尝试“pickle”来保存模型。 请尝试以下步骤保存模型。在

import pickle
with open('<model_name.pkl>', 'wb') as f:
    pickle.dump(<your_model>, f)

将模型解读为：

^{pr2}$

在您的例子中，对于版本1，代码片段中的模型是svmod2。在