当前位置: 首页 > 文档资料 > Caffe 学习笔记 >

caffe2 概述

优质
小牛编辑
132浏览
2023-12-01

Caffe原作者贾杨清试验性重构了Caffe,称为Caffe2

Caffe2出发点是改善Caffe的设计,使之更通用,不局限于计算机视觉领域,而是更广泛的机器学习任务

改进

设备支持

以前是这样的:

void Forward_cpu();
void Forward_gpu();
void Backward_cpu();
void Backward_gpu();

现在改为更简洁的接口:

template<typename Device> class Layer
{
    void Forward();
    void Backward();
}

Blob

以前是这样的:

template<typename Dtype> class Blob
{
    Dtype *data;
    ...
}

现在改为:

class Blob
{
    AnyPointer data;
    DataType type;
}