Caffe2 - 定义复杂网络


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

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

概括

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