ModelArts模型管理导入FrozenGraph格式模型
【摘要】 模型管理 模型导入 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格式的模型就可以了。
![1604994471269079422.png image.png](https://bbs-img.huaweicloud.com/blogs/img/1604994471269079422.png)
推荐
![](https://res.hc-cdn.com/ecology/9.3.130/v2_resources\images\global\blog-dialog.png)
华为开发者空间发布
让每位开发者拥有一台云主机
【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)