请问老师,训练好的卷积神经网络怎么保存下来。不用每次都需要训练吧。

来源:4-21 动手实现CNN卷积神经网络(三)

helloisaac

2018-12-25

请问老师,训练好的卷积神经网络怎么保存下来。不用每次都需要训练吧。
是不是有对训练好的神经网络的保存和加载方式?

写回答

1回答

Oscar

2018-12-25

当然。用 tf.train.Saver 类的 save 和 restore 方法分别可以保存和加载保存的模型文件。

我后面的例子程序有给到使用代码的,你可以搜一下。

看很完整的案例请参考:

Tensorflow加载预训练模型和保存模型

0
1
helloisaac
非常感谢!
2018-12-30
共1条回复

基于Python玩转人工智能最火框架 TensorFlow应用实践

机器学习入门,打牢TensorFlow框架应用是关键!

2214 学习 · 688 问题

查看课程