Python Layer

The Python layer allows users to add customized layers without modifying the Caffe core code.


message PythonParameter {
  optional string module = 1;
  optional string layer = 2;
  // This value is set to the attribute `param_str` of the `PythonLayer` object
  // in Python before calling the `setup()` method. This could be a number,
  // string, dictionary in Python dict format, JSON, etc. You may parse this
  // string in `setup` method and use it in `forward` and `backward`.
  optional string param_str = 3 [default = ''];
  optional bool share_in_parallel = 4 [default = false];

Examples and tutorials

Dialogue & Discussion