Linux 拨号vps windows公众号手机端

Cafe2框架中怎么保存和加载已训练的模型

lewis 5年前 (2020-01-19) 阅读数 10 #大数据
文章标签 Cafe2

在Cafe2框架中,可以使用caffe2的SaveLoad函数来保存和加载已训练的模型。

保存模型:

# Save the trained model
workspace.SaveNet("path/to/model.pb", "path/to/model_init.pb")

加载模型:

# Load the trained model
init_net, predict_net = workspace.LoadNet("path/to/model_init.pb", "path/to/model.pb")

通过这种方式,您可以保存已训练的模型并在需要时加载它。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门