【发布时间】:2023-01-24 16:58:52
【问题描述】:
我正在处理此链接https://keras.io/examples/vision/handwriting_recognition/ 上的代码
它工作正常但是当我训练模型并使用 model.save() 保存它时,当我在另一个程序中使用 tf.keras.models.load_model() 加载它时等于原始但没有训练阶段(在这种情况下指令model.fit()),推理预期结果不匹配。
Expected result (like the Keras Page)
这不是保存格式的问题,因为 SavedModel 和 h5 格式仍然存在问题。
保存
history = model.fit(
train_ds,
validation_data=validation_ds,
epochs=epochs,
callbacks=[edit_distance_callback],
)
model.save(path)
然后程序结束。
加载中
model = keras.models.load_model(path)
然后我像教程一样进行推理。
【问题讨论】:
标签: tensorflow keras serialization handwriting-recognition