无法将PB模型加载到keras模型中

时间:2020-07-15 12:15:33

标签: tensorflow keras

我有一个预先训练的pb模型,我试图将其作为keras模型加载,这样我就可以运行诸如model.summary()之类的东西,但是加载pb模型之后的类型是“ AutoTrackable”对象,因为该对象没有属性“摘要”。这是我正在使用的代码

model = tf.saved_model.load('my_model/') # folder that contains the pb model
model.summary()

,这是错误消息:

"AttributeError: 'AutoTrackable' object has no attribute 'summary'"

有人可以帮忙吗?

0 个答案:

没有答案