gpt4 book ai didi

python-3.x - 保存包裹在 Keras 中的模型

转载 作者:行者123 更新时间:2023-12-05 07:13:54 26 4
gpt4 key购买 nike

很抱歉我的问题很幼稚,但我正在尝试保存我的 keras 模型 (),我在其中使用 TFBertModel() 函数作为隐藏层。为此,我使用了 tf.keras 包提供的 save() 函数。

但是我得到了这个错误:

---------------------------------------------------------------------------

NotImplementedError Traceback (most recent call last)

<ipython-input-13-3b315f7219da> in <module>()
----> 1 model.save('model_weights.h5')

8 frames

/tensorflow-2.1.0/python3.6/tensorflow_core/python/keras/engine/network.py in get_config(self)
915 def get_config(self):
916 if not self._is_graph_network:
--> 917 raise NotImplementedError
918 return copy.deepcopy(get_network_config(self))
919

NotImplementedError:

错误可以从我的 colab 重现:https://colab.research.google.com/drive/18HYwffkXCylPqeA-8raL82vfwOjb-aLP

另一个问题是我应该如何调用这个模型进行预测?

感谢您的帮助!

最佳答案

关于python-3.x - 保存包裹在 Keras 中的模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60062624/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com