如何用可训练参数编写caffe python层?

时间:2016-01-27 01:18:14

标签: deep-learning caffe pycaffe

我想学习如何编写caffe python图层 但我只找到关于非常简单的图层的示例,例如pyloss

如何用可训练的参数编写python caffe?

例如,如何编写完全连接的python层?

1 个答案:

答案 0 :(得分:1)

Caffe将图层的可训练参数存储为blobs的向量。默认情况下,此向量为空,您可以在图层的setup中向其添加参数blob。对于参数在test_python_layer.py中的图层,有一个简单示例。

有关caffe。{/ p>中"Python"图层的详细信息,请参阅this post