标签: python tensorflow
我使用多种不同格式保存了相同的模型,一种使用标准model.save()使用来自tensorflow的SavedModel格式,另一种使用了model.save('*。h5')使用hdf5格式。 / p>
在加载模型进行推理时,尽管磁盘上的实际模型只有5mb(.pb文件),变量/资产只有几kb,但SavedModel文件会将1Gb加载到内存中。但是h5文件的内存只有大约50mb。
我真的不确定为什么SavedModel方法会占用这么多的内存并在寻找答案。这种方法不可用吗?