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

lewis 2020-01-19 23次阅读

在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")

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



发表评论:

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