forked from kevinlin311tw/Caffe-DeepBinaryCode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathinsert_splits.hpp
More file actions
26 lines (17 loc) · 833 Bytes
/
insert_splits.hpp
File metadata and controls
26 lines (17 loc) · 833 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#ifndef _CAFFE_UTIL_INSERT_SPLITS_HPP_
#define _CAFFE_UTIL_INSERT_SPLITS_HPP_
#include <string>
#include "caffe/proto/caffe.pb.h"
namespace caffe {
// Copy NetParameters with SplitLayers added to replace any shared bottom
// blobs with unique bottom blobs provided by the SplitLayer.
void InsertSplits(const NetParameter& param, NetParameter* param_split);
void ConfigureSplitLayer(const string& layer_name, const string& blob_name,
const int blob_idx, const int split_count, const float loss_weight,
LayerParameter* split_layer_param);
string SplitLayerName(const string& layer_name, const string& blob_name,
const int blob_idx);
string SplitBlobName(const string& layer_name, const string& blob_name,
const int blob_idx, const int split_idx);
} // namespace caffe
#endif // CAFFE_UTIL_INSERT_SPLITS_HPP_