- 博客(4)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 Caffe2填坑系列(4)----指定梯度计算时的输入和输出
class GetInterpGradient ： public GradientMakerBase{ using GradientMakerBase::GradientMakerBase; vector<OperatorDef> GetGradientDefs() override{ return SingleGradientDef( ...
2019-04-29 23:43:56
349
原创 Caffe2填坑系列(3)----OPERATOR_SCHEMA()
operator_schema.hNumInputs()// Functions to set the property of the operator schemas. // Sets the number of inputs, either a fixed number or a min and a max. /** * @brief A single in...
2019-04-25 23:34:41
307
原创 Caffe2填坑系列(2)----获取数据或参数的方法
pad_end_(OperatorBase::template GetSingleArgument(“pad_end”,0))如果参数“pad_end”没有提供，则采用0为默认值获取Op的输入和输出const auto& X = Input(0)：获取第一个输入，注意输入一般为常量类型，因为我们一般只读不更改数据。auto* Y = Output(0): 指向第一个输出，不能为常...
2019-04-24 22:23:21
322
原创 Caffe2填坑系列(1)----三个头文件
已经有将近一年的时间没有写博客了。过去的一年了，学习了很多新的东西，在这里和大家分享一下，希望对你有所帮助。特别声明：我在这个系列中所使用的Caffe2，不是已经合并进Pytorch的Caffe2，而是为合并前的。这两者的差别还是很大的。今天，作为第一讲，内容可以说非常少。- #include "caffe2/core/context.h"选择是CPU或者GPU- #include "...
2019-04-19 23:29:18
372
Caffe-Windows下画loss和accuracy曲线python代码
2016-12-09
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人