我想用10倍简历将经过训练的模型保存在keras
中,并且我在模型中使用了词嵌入,因此keras
保存的模型文件太大,大约550M。 10倍训练的模型约为5GB。
如果我可以在保存的模型中删除嵌入变量,而只保存另一个变量,我想我可以节省大多数文件大小,因为单词“嵌入数组文件”大约为500M。然后,文件总大小将减少到1GB。
但是我无法使用keras
来实现。
此外,当我尝试在keras
中使用tf.train.saver时,会发生一些奇怪的事情。
有人有什么主意吗?
答案 0 :(得分:0)
您是否只尝试保存权重,而不是此处所述的整个模型:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model
model.save_weights('my_model_weights.h5')