如何在caffe中添加自定义的激活函数
1个回答
展开全部
下面是一个官方例子, 自定义了sin激活函数:
BVLC/caffe
总体上讲, 包括下面几步:
写hpp, cpp(CPU上运行), cu(GPU上运行)文件
(可选)写测试文件
make重新编译caffe
(可选)make test重新编译测试文件和make runtest进行测试
之后就可以像用ReLU一样用你的激活函数了.
BVLC/caffe
总体上讲, 包括下面几步:
写hpp, cpp(CPU上运行), cu(GPU上运行)文件
(可选)写测试文件
make重新编译caffe
(可选)make test重新编译测试文件和make runtest进行测试
之后就可以像用ReLU一样用你的激活函数了.
已赞过
已踩过<
评论
收起
你对这个回答的评价是？
推荐律师服务：
若未解决您的问题，请您详细描述您的问题，通过百度律临进行免费专业咨询
