nvcaffe
文章平均质量分 92
KFXW
HUST
展开
-
NVcaffe源码阅读——Layer
nvcaffe的Layer.hpp/cpp在功能上与以前的caffe相比有一些明显改变。nvcaffe对Layer.hpp的代码进行了拓展,提供了很多便利的新特性。原创 2017-07-27 00:14:20 · 2190 阅读 · 0 评论 -
NVcaffe源码阅读——Net&Solver
在caffe中,为了区分多GPU数据并行时,负责更新网络参数的solver被称作root solver。 其他仅仅计算梯度用的solver为WorkerSolver,是Solver类的子类。 nvcaffe将WorkerSolver类同Solver本身合并,将root solver和worker solver的概念融入到统一的框架下去,添加了is_root()等函数,也添加了Reduce()等用于并行处理的函数。原创 2017-07-31 18:27:20 · 1566 阅读 · 0 评论 -
NVcaffe源码阅读——Blob的重新构建
nvcaffe由于对数据类型做了很大的拓展,所以作者对blob的构建上做了很大改动。即向下的内存/显存管理提供了tensor类,blob类使用tensor类作为成员变量代替以前的data和diff指针;blob提供与caffe相兼容的api借口,保持功能的一致性,添加部分适应新功能的api;最后使用Tblob类包裹blob类,提供更加灵活的模板定义。原创 2017-07-26 22:58:17 · 2965 阅读 · 0 评论