reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-17 f7c4a3cfd07adede3308f8d9d3d7315427d90a7c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#ifndef CAFFE2_OPT_OPTIMIZER_H
#define CAFFE2_OPT_OPTIMIZER_H
 
#include "caffe2/core/common.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/workspace.h"
 
namespace caffe2 {
namespace opt {
 
CAFFE2_API NetDef optimize(NetDef net, Workspace* ws, int level = 1);
CAFFE2_API NetDef optimize(NetDef net, int level = 1);
 
} // namespace opt
} // namespace caffe2
 
#endif // CAFFE2_OPT_OPTIMIZER_H