为什么我们在深度学习中使用 Pickle 而不是以 HDF5 格式保存权重

数据挖掘 机器学习 泡菜
2022-03-05 21:23:16

我已经读过,Pickle 库用于将训练好的模型保存到.pkl文件中以供以后使用。此外,我们可以使用以下方法将训练模型的权重保存为 hdf5 格式

model.save()

然后使用这些权重。

那么为什么我们使用 pickle 而不是 hdf5 文件呢?

2个回答

避免泡菜,有几个问题

即使在 keras 文档中,他们也说

“不建议使用 pickle 或 cPickle 来保存 Keras 模型。”

另请查看此解决方案。

截至 2021 年 12 月,既不推荐 pickle 也不推荐 h5(而 Keras/TF 仍支持 h5)。

文档说

您可以使用两种格式将整个模型保存到磁盘:TensorFlow SavedModel 格式和较旧的 Keras H5 格式。推荐的格式是 SavedModel。当您使用 model.save() 时,它是默认值