ModelArts模型管理导入FrozenGraph格式模型

当前ModelArts文档中只有SavedModel格式的案例

ModelArts模型导入事实上也支持使用FrozenGraph进行模型导入。

通过阅读文档https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0093.html

从文档中可以看到有如下方法进行重写

那么这两种模型导入的差别在哪里呢。

如果导入的模型文件是SavedModel，那么一般情况下是不需要 实现__init__(self,model_name,model_path)方法。

但是如果导入的模型是FrozenGraph就需要实现__init__(self,model_name,model_path)方法，并且在此方法中实现模型的加载。

以tensorflow2.0为例

https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0301.html

__init__(self,model_name,model_path)方法中调用了load_model，因此只要在load_model()方法中实现加载FozenGraph格式的模型就可以了。

AI开发平台ModelArts

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。