src/caffeine/proto/layer_param.proto

   1 package caffeine;
   2
   3 message BlobProto {
   4   optional int32 num = 1 [default = 0];
   5   optional int32 height = 2 [default = 0];
   6   optional int32 width = 3 [default = 0];
   7   optional int32 channels = 4 [default = 0];
   8   repeated float data = 5;
   9   repeated float diff = 6;
  10 }
  11
  12 message FillerParameter {
  13   // The filler type. In default we will set it to Gaussian for easy
  14   // debugging.
  15   required string type = 1 [default = 'gaussian'];
  16   optional float value = 2 [default = 0]; // the value in constant filler
  17   optional float min = 3 [default = 0]; // the min value in uniform filler
  18   optional float max = 4 [default = 1]; // the max value in uniform filler
  19   optional float mean = 5 [default = 0]; // the mean value in gaussian filler
  20   optional float std = 6 [default = 1]; // the std value in gaussian filler
  21 }
  22
  23 message LayerParameter {
  24   required string name = 1; // the layer name
  25   required string type = 2; // the string to specify the layer type
  26
  27   // Parameters to specify layers with inner products.
  28   optional int32 num_output = 3; // The number of outputs for the layer
  29   optional bool biasterm = 4 [default = true]; // whether to have bias terms
  30   optional FillerParameter weight_filler = 5; // The filler for the weight
  31   optional FillerParameter bias_filler = 6; // The filler for the bias
  32
  33   optional uint32 pad = 7 [default = 0]; // The padding size
  34   optional float pad_value = 8 [default = 0]; // The padding value
  35   optional uint32 kernelsize = 9; // The kernel size
  36   optional uint32 group = 10 [default = 1]; // The group size for group conv
  37   optional uint32 stride = 11 [default = 1]; // The stride
  38   optional string pool = 12 [default = 'max']; // The pooling method
  39   optional float dropout_ratio = 13 [default = 0.5]; // dropout ratio
  40
  41   optional float alpha = 14 [default = 1.]; // for local response norm
  42   optional float beta = 15 [default = 0.75]; // for local response norm
  43 }