我怎样才能开始为我的图层编写代码?
How can I start writing the code for my layer?
我看到研究人员正在为 Caffe 的原始版本添加一些功能,并根据需要使用这些层和功能,然后通过 Github
共享这些版本。如果我没记错的话,有两种方法: 1)
添加c++和Cuda版本的层后重新编译Caffe。 2)
为该功能编写 python 代码,并在 Caffe 中将其称为 python 层。
我想根据我的研究问题向Caffe 添加一个新层。我真的不知道我应该从哪个点开始写新层,我应该考虑哪些步骤。
我的问题是:
1)有没有什么文档或者学习资料可以用来写层?
2) 上述添加新层的方法中,哪种方式更受欢迎?
非常感谢任何帮助和指导
非常感谢
出于研究目的,对于"playing around",写一个python层通常更方便:省去了编译等工作
您可以在 "Python"
图层 .
上找到简短教程
另一方面,如果您想要更好的性能,您应该为您的层编写本机 C++ 代码。
你可以找到关于它的简短解释 here.
我看到研究人员正在为 Caffe 的原始版本添加一些功能,并根据需要使用这些层和功能,然后通过 Github
共享这些版本。如果我没记错的话,有两种方法: 1)
添加c++和Cuda版本的层后重新编译Caffe。 2)
为该功能编写 python 代码,并在 Caffe 中将其称为 python 层。
我想根据我的研究问题向Caffe 添加一个新层。我真的不知道我应该从哪个点开始写新层,我应该考虑哪些步骤。
我的问题是:
1)有没有什么文档或者学习资料可以用来写层?
2) 上述添加新层的方法中,哪种方式更受欢迎?
非常感谢任何帮助和指导
非常感谢
出于研究目的,对于"playing around",写一个python层通常更方便:省去了编译等工作
您可以在 "Python"
图层
另一方面,如果您想要更好的性能,您应该为您的层编写本机 C++ 代码。
你可以找到关于它的简短解释 here.