微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何将在tensorflow 2中训练的模型转换为tensorflow 1冻结图

如何解决如何将在tensorflow 2中训练的模型转换为tensorflow 1冻结图

我想使用tensorflow 2训练模型,但是之后我需要使用仅与tensorflow 1兼容的转换器。是否可以,如果可以,我如何将使用tensorflow 2训练的模型转换为tensorflow 1种格式?

解决方法

如果没有可靠地将TF2模型转换为TF1的方法,则始终可以保存经过训练的参数(权重,偏差),并在以后使用它们来初始化TF1图。 我以前是出于其他目的这样做的。您可以保存如下:

weights = []
for layer in model.layers:

    w = layer.get_weights()
    if len(w)>0:
      print(layer.name)
      weights.append(w)


with open('mnist_weights.pkl','wb') as f:
  pickle.dump(weights,f)  

w[0]=weightsw[1]=biases每一层的位置

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。