Tensorflow SavedModel方法比H5大得多

时间:2019-12-11 00:26:05

标签: python tensorflow

我使用多种不同格式保存了相同的模型,一种使用标准model.save()使用来自tensorflow的SavedModel格式,另一种使用了model.save('*。h5')使用hdf5格式。 / p>

在加载模型进行推理时,尽管磁盘上的实际模型只有5mb(.pb文件),变量/资产只有几kb,但SavedModel文件会将1Gb加载到内存中。但是h5文件的内存只有大约50mb。

我真的不确定为什么SavedModel方法会占用这么多的内存并在寻找答案。这种方法不可用吗?

0 个答案:

没有答案