Caffe2 - 定义复杂网络

在上一课中,您学习了如何创建一个简单的网络,以及如何执行它并检查其输出。创建复杂网络的过程与上面描述的过程类似。Caffe2 提供了大量用于创建复杂架构的运算符。我们鼓励您查看 Caffe2 文档以获取运算符列表。在研究了各种运算符的用途后,您就可以创建复杂网络并对其进行训练了。为了训练网络,Caffe2 提供了几个预定义计算单元 - 即运算符。您需要选择适当的运算符来训练您的网络,以解决您尝试解决的问题类型。

一旦网络训练到您满意的程度,您就可以将其存储在类似于您之前使用的预训练模型文件的模型文件中。这些训练过的模型可能会被贡献给 Caffe2 存储库,以造福其他用户。或者,您也可以简单地将训练好的模型用于您自己的私人生产用途。

摘要

Caffe2 是一个深度学习框架,它允许您尝试使用多种神经网络来预测数据。Caffe2 网站提供了许多预先训练的模型。您学会了使用其中一个预先训练的模型对给定图像中的对象进行分类。您还学会了定义您选择的神经网络架构。可以使用 Caffe 中的许多预定义运算符来训练此类自定义网络。训练好的模型存储在一个文件中,可以将其带入生产环境。