Sunshine_in_Moon的专栏

天行健君子以自强不息!

Caffe2填坑系列(4)----指定梯度计算时的输入和输出

class GetInterpGradient : public GradientMakerBase{ using GradientMakerBase::GradientMakerBase; vector<OperatorDef> GetGradientDe...

2019-04-29 23:43:56

阅读数 26

评论数 0

Caffe2填坑系列(3)----OPERATOR_SCHEMA()

operator_schema.h NumInputs() // Functions to set the property of the operator schemas. // Sets the number of inputs, either a fixed number or ...

2019-04-25 23:34:41

阅读数 56

评论数 0

Caffe2填坑系列(2)----获取数据或参数的方法

pad_end_(OperatorBase::template GetSingleArgument(“pad_end”,0)) 如果参数“pad_end”没有提供,则采用0为默认值 获取Op的输入和输出 const auto& X = Input(0):获取第一个输入,注意输入一般...

2019-04-24 22:23:21

阅读数 14

评论数 0

Caffe2填坑系列(1)----三个头文件

已经有将近一年的时间没有写博客了。过去的一年了,学习了很多新的东西,在这里和大家分享一下,希望对你有所帮助。 特别声明:我在这个系列中所使用的Caffe2,不是已经合并进Pytorch的Caffe2,而是为合并前的。这两者的差别还是很大的。 今天,作为第一讲,内容可以说非常少。 - #incl...

2019-04-19 23:29:18

阅读数 20

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭