Keras如何保存指定的变量?

时间:2018-06-29 14:17:23

标签: tensorflow machine-learning keras nlp deep-learning

我想用10倍简历将经过训练的模型保存在keras中,并且我在模型中使用了词嵌入,因此keras保存的模型文件太大,大约550M。 10倍训练的模型约为5GB。

如果我可以在保存的模型中删除嵌入变量,而只保存另一个变量,我想我可以节省大多数文件大小,因为单词“嵌入数组文件”大约为500M。然后,文件总大小将减少到1GB。

但是我无法使用keras来实现。

此外,当我尝试在keras中使用tf.train.saver时,会发生一些奇怪的事情。

有人有什么主意吗?

1 个答案:

答案 0 :(得分:0)

您是否只尝试保存权重,而不是此处所述的整个模型:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

model.save_weights('my_model_weights.h5')