深度学习框架
关注
分享
扫一扫
hnshahao
这个作者很懒，什么都没留下…
展开
-
Pytorch (1) - 模型的格式
Pytorch 模型的格式原创 2018-07-06 11:06:30 · 1238 阅读 · 0 评论 -
Caffe (1) Blob 层代码解读
Caffe 中所有的数据都是存放在Blob对象中，废话不多说，直接上代码，头文件 实际上blob是对SyncedMemory对象做了一次封装，意思是在内部，做了几个SyncedMemory类型变量指针，对于CPU和GPU内存管理最为核心的部分是在SyncedMemory对象中，后续将详细介绍这个部分。 （1）头文件 #ifndef CAFFE_BLOB_HPP_ #define CAFFE...原创 2018-07-26 13:36:49 · 469 阅读 · 0 评论 -
Caffe (2) SyncedMemory内存管理机制
在Caffe中，blob是对于上层空间的数据管理存储对象，对于上层来说的话，大部分时候是直接取blob对象的指针来用，如果不考虑GPU的情况下，实际上很简单，就是返回指针就行，但是问题是通常的数据是在GPU和CPU上同时存在，需要两个数据在不同的设备上进行同步，那么SyncedMemory的作用是实际上在管理实际数据。对于Blob中，封装的3个SyncedMemory对象的智能指针: 【大的逻辑...原创 2018-07-26 15:44:20 · 1127 阅读 · 0 评论