您的位置:首页 > 移动开发

深度学习Matlab工具箱代码注释——cnnapplygrads.m

2017-07-24 09:58 691 查看
%%=========================================================================
%函数名称:cnnapplygrads(),权值更新函数
%输入参数:net,权值待更新的卷积神经网络;opts,神经网络训练的相关参数
%输出参数:
%算法流程:先更新卷积层的参数,再更新全连接层参数
%注意事项:
%%=========================================================================
function net = cnnapplygrads(net, opts)
for l = 2 : numel(net.layers)
if strcmp(net.layers{l}.type, 'c')
for j = 1 : numel(net.layers{l}.a)
for ii = 1 : numel(net.layers{l - 1}.a)

%这里没什么好说的,就是普通的权值更新的公式:W_new = W_old - alpha * de/dW(误差对权值导数)
net.layers{l}.k{ii}{j} = net.layers{l}.k{ii}{j} - opts.alpha * net.layers{l}.dk{ii}{j};
end
net.layers{l}.b{j} = net.layers{l}.b{j} - opts.alpha * net.layers{l}.db{j};
end
end
end

net.ffW = net.ffW - opts.alpha * net.dffW;
net.ffb = net.ffb - opts.alpha * net.dffb;
end


原文链接
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  深度学习 matlab