使用自定义图层保存Tensorflow模型

时间:2020-11-07 02:37:34

标签: python tensorflow model save

我通读了文档,但对我来说还不清楚:如果我编写了一个自定义层,然后在模型中使用它,我是否可以将模型另存为SavedModel,而自定义层会自动进入其中或执行我也必须保存自定义图层吗? 我尝试仅将模型保存为H5格式,而不保存自定义层。尝试加载模型时,自定义层出现错误,无法识别或类似错误。通读文档,我发现将自定义对象保存为H5格式要花点时间。但是,它如何与SavedModels一起使用?

0 个答案:

没有答案