最笨的方法(3)
solver
计算梯度并实现参数更新由Solver::ComputeUpdateValue()完成,它用残差梯度和权重衰减一起得到最终梯度,当然是利用当前权值计算得到,最终的update存在blob类的diff中,Blob::Update方法做最终的update,到底谁做?
layers
vision layers
指以矩形块为输入和输出的layer,考虑像素的空间位置。
1.卷积
有CPU实现和GPU实现两个版本(为了用显卡计算要用cuda代码,数据层需要cuda版本的实现吗?不用的话如何把数据读到显存内?写的时候照着row-major索引来写,不要用指针,cuda不支持指针)

浙公网安备 33010602011771号