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;
}