在Caffe中编写具有可学习参数的自定义Python层

Question

3

我知道这个示例旨在演示如何在Python layer中使用add_blob()方法添加可训练参数。

然而，我仍然无法理解如何基于用户定义的参数设置blob的尺寸。

这里有一个更好的示例来编写一个带有 trainable 参数的 Python layer。这里但是，在这里，该层不包含任何可训练参数。

请说明如何编写具有可训练参数的自定义 Python 层。

- Suhas Lohit

我不确定你所说的“设置blob的尺寸”是什么意思。据我所知，Caffe会根据底部blob的大小和当前层的属性来设置顶部blob的大小，因此您不需要设置尺寸。 - Amir

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Shai · Answer 1

当您使用add_blob()添加参数blob时，您可以在setup()方法中（在添加时）或在层的reshape()方法中重新调整添加的blob。