reid from https://github.com/michuanhaohao/reid-strong-baseline
554325746@qq.com
2020-02-05 7c6a958f00e327a8991f48093456de7cb84cf918
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