| layer { | |
| name: "data" | |
| type: "HDF5Data" | |
| top: "data" | |
| top: "label" | |
| hdf5_data_param { | |
| source: "examples/hdf5_classification/data/train.txt" | |
| batch_size: 10 | |
| } | |
| } | |
| layer { | |
| name: "ip1" | |
| type: "InnerProduct" | |
| bottom: "data" | |
| top: "ip1" | |
| inner_product_param { | |
| num_output: 40 | |
| weight_filler { | |
| type: "xavier" | |
| } | |
| } | |
| } | |
| layer { | |
| name: "relu1" | |
| type: "ReLU" | |
| bottom: "ip1" | |
| top: "ip1" | |
| } | |
| layer { | |
| name: "ip2" | |
| type: "InnerProduct" | |
| bottom: "ip1" | |
| top: "ip2" | |
| inner_product_param { | |
| num_output: 2 | |
| weight_filler { | |
| type: "xavier" | |
| } | |
| } | |
| } | |
| layer { | |
| name: "accuracy" | |
| type: "Accuracy" | |
| bottom: "ip2" | |
| bottom: "label" | |
| top: "accuracy" | |
| } | |
| layer { | |
| name: "loss" | |
| type: "SoftmaxWithLoss" | |
| bottom: "ip2" | |
| bottom: "label" | |
| top: "loss" | |
| } | |