model.save() 保存模型时报错
Opened this issue · 1 comments
model.save() 保存SavedModel时,报错:
Traceback (most recent call last):
File "/Users/luyafei/Downloads/code/DeepTextMatch/SBERT/run.py", line 142, in
model.fit(train_generator.forfit(),
File "/Users/luyafei/opt/anaconda3/envs/py3-tf2-new/lib/python3.9/site-packages/keras/utils/traceback_utils.py", line 67, in error_handler
raise e.with_traceback(filtered_tb) from None
File "/Users/luyafei/opt/anaconda3/envs/py3-tf2-new/lib/python3.9/site-packages/keras/layers/serialization.py", line 117, in populate_deserializable_objects
LOCAL.ALL_OBJECTS['Functional'] = models.Functional
AttributeError: module 'keras.api._v2.keras.models' has no attribute 'Functional'
提问时请尽可能提供如下信息:
基本信息
- 你使用的操作系统: mac linux下都尝试不行
- 你使用的Python版本: 3.9
- 你使用的Tensorflow版本: tf2.9.0
- 你使用的Keras版本: 2.9.0
- 你使用的bert4keras版本: v0.11.1
- 你使用纯keras还是tf.keras: tf.keras
- 你加载的预训练模型: chinese_wwm_ext_L-12_H-768_A-12
核心代码
# 6. model fit
model.fit(train_generator.forfit(),
steps_per_epoch=len(train_generator),
epochs=epochs,
callbacks=[earlystop, checkpoint])
model.save("../output/savedmodel/") # SavedModel 格式
输出信息
# 请在此处贴上你的调试输出
自我尝试
不管什么问题,请先尝试自行解决,“万般努力”之下仍然无法解决再来提问。此处请贴上你的努力过程。
这么新的tf和keras我无能为力啊,抱歉。