optional bool normalize = 2 [default = true];
}
-// Message that stores parameters used by AccuracyLayer
+// Messages that store parameters used by individual layer types follow, in
+// alphabetical order.
+
message AccuracyParameter {
// When computing accuracy, count as correct by comparing the true label to
// the top k scoring classes. By default, only compare to the top scoring
optional int32 ignore_label = 3;
}
-// Message that stores parameters used by ArgMaxLayer
message ArgMaxParameter {
// If true produce pairs (argmax, maxval)
optional bool out_max_val = 1 [default = false];
optional uint32 top_k = 2 [default = 1];
}
-// Message that stores parameters used by ConcatLayer
message ConcatParameter {
// The axis along which to concatenate -- may be negative to index from the
// end (e.g., -1 for the last axis). Other axes must have the
optional uint32 concat_dim = 1 [default = 1];
}
-// Message that stores parameters used by ContrastiveLossLayer
message ContrastiveLossParameter {
// margin for dissimilar pair
optional float margin = 1 [default = 1.0];
optional bool legacy_version = 2 [default = false];
}
-// Message that stores parameters used by ConvolutionLayer
message ConvolutionParameter {
optional uint32 num_output = 1; // The number of outputs for the layer
optional bool bias_term = 2 [default = true]; // whether to have bias terms
optional Engine engine = 15 [default = DEFAULT];
}
-// Message that stores parameters used by DataLayer
message DataParameter {
enum DB {
LEVELDB = 0;
optional bool force_encoded_color = 9 [default = false];
}
-// Message that stores parameters used by DropoutLayer
message DropoutParameter {
optional float dropout_ratio = 1 [default = 0.5]; // dropout ratio
}
-// Message that stores parameters used by DummyDataLayer.
// DummyDataLayer fills any number of arbitrarily shaped blobs with random
// (or constant) data generated by "Fillers" (see "message FillerParameter").
message DummyDataParameter {
repeated uint32 width = 5;
}
-// Message that stores parameters used by EltwiseLayer
message EltwiseParameter {
enum EltwiseOp {
PROD = 0;
optional bool stable_prod_grad = 3 [default = true];
}
-// Message that stores parameters used by ExpLayer
message ExpParameter {
// ExpLayer computes outputs y = base ^ (shift + scale * x), for base > 0.
// Or if base is set to the default (-1), base is set to e,
optional float shift = 3 [default = 0.0];
}
-// Message that stores parameters used by HDF5DataLayer
message HDF5DataParameter {
// Specify the data source.
optional string source = 1;
optional bool shuffle = 3 [default = false];
}
-// Message that stores parameters used by HDF5OutputLayer
message HDF5OutputParameter {
optional string file_name = 1;
}
optional Norm norm = 1 [default = L1];
}
-// Message that stores parameters used by ImageDataLayer
message ImageDataParameter {
// Specify the data source.
optional string source = 1;
optional string root_folder = 12 [default = ""];
}
-// Message that stores parameters InfogainLossLayer
message InfogainLossParameter {
// Specify the infogain matrix source.
optional string source = 1;
}
-// Message that stores parameters used by InnerProductLayer
message InnerProductParameter {
optional uint32 num_output = 1; // The number of outputs for the layer
optional bool bias_term = 2 [default = true]; // whether to have bias terms
optional int32 axis = 5 [default = 1];
}
-// Message that stores parameters used by LRNLayer
message LRNParameter {
optional uint32 local_size = 1 [default = 5];
optional float alpha = 2 [default = 1.];
optional float k = 5 [default = 1.];
}
-// Message that stores parameters used by MemoryDataLayer
message MemoryDataParameter {
optional uint32 batch_size = 1;
optional uint32 channels = 2;
optional uint32 width = 4;
}
-// Message that stores parameters used by MVNLayer
message MVNParameter {
// This parameter can be set to false to normalize mean only
optional bool normalize_variance = 1 [default = true];
optional float eps = 3 [default = 1e-9];
}
-// Message that stores parameters used by PoolingLayer
message PoolingParameter {
enum PoolMethod {
MAX = 0;
optional bool global_pooling = 12 [default = false];
}
-// Message that stores parameters used by PowerLayer
message PowerParameter {
// PowerLayer computes outputs y = (shift + scale * x) ^ power.
optional float power = 1 [default = 1.0];
optional float shift = 3 [default = 0.0];
}
-// Message that stores parameters used by PythonLayer
message PythonParameter {
optional string module = 1;
optional string layer = 2;
}
-// Message that stores parameters used by ReLULayer
message ReLUParameter {
// Allow non-zero slope for negative inputs to speed up optimization
// Described in:
optional Engine engine = 2 [default = DEFAULT];
}
-// Message that stores parameters used by ReshapeLayer
message ReshapeParameter {
// Specify the output dimensions. If some of the dimensions are set to 0,
// the corresponding dimension from the bottom layer is used (unchanged).
optional int32 num_axes = 3 [default = -1];
}
-// Message that stores parameters used by SigmoidLayer
message SigmoidParameter {
enum Engine {
DEFAULT = 0;
optional Engine engine = 1 [default = DEFAULT];
}
-// Message that stores parameters used by SliceLayer
message SliceParameter {
// The axis along which to slice -- may be negative to index from the end
// (e.g., -1 for the last axis).
optional int32 axis = 2 [default = 1];
}
-// Message that stores parameters used by TanHLayer
message TanHParameter {
enum Engine {
DEFAULT = 0;
optional Engine engine = 1 [default = DEFAULT];
}
-// Message that stores parameters used by ThresholdLayer
message ThresholdParameter {
optional float threshold = 1 [default = 0]; // Strictly positive values
}
-// Message that stores parameters used by WindowDataLayer
message WindowDataParameter {
// Specify the data source.
optional string source = 1;
optional string root_folder = 13 [default = ""];
}
-// Message that stores parameters used by SPPLayer
message SPPParameter {
enum PoolMethod {
MAX = 0;
optional HDF5OutputParameter hdf5_output_param = 1001;
}
-// Message that stores parameters used by PReLULayer
message PReLUParameter {
// Parametric ReLU described in K. He et al, Delving Deep into Rectifiers:
// Surpassing Human-Level Performance on ImageNet Classification, 2015.
projects / platform / upstream / caffeonacl.git / commitdiff