| name: 'LinearRegressionExample' |
| |
| |
| layer { |
| type: 'DummyData' |
| name: 'x' |
| top: 'x' |
| dummy_data_param { |
| shape: { dim: 10 dim: 3 dim: 2 } |
| data_filler: { type: 'gaussian' } |
| } |
| } |
| layer { |
| type: 'DummyData' |
| name: 'y' |
| top: 'y' |
| dummy_data_param { |
| shape: { dim: 10 dim: 3 dim: 2 } |
| data_filler: { type: 'gaussian' } |
| } |
| } |
| |
| |
| layer { |
| type: 'InnerProduct' |
| name: 'ipx' |
| top: 'ipx' |
| bottom: 'x' |
| inner_product_param { |
| num_output: 10 |
| weight_filler { type: 'xavier' } |
| } |
| } |
| layer { |
| type: 'InnerProduct' |
| name: 'ipy' |
| top: 'ipy' |
| bottom: 'y' |
| inner_product_param { |
| num_output: 10 |
| weight_filler { type: 'xavier' } |
| } |
| } |
| layer { |
| type: 'Python' |
| name: 'loss' |
| top: 'loss' |
| bottom: 'ipx' |
| bottom: 'ipy' |
| python_param { |
| |
| module: 'pyloss' |
| |
| layer: 'EuclideanLossLayer' |
| } |
| |
| |
| |
| loss_weight: 1 |
| } |
|
|