55 lines
782 B
Plaintext
55 lines
782 B
Plaintext
layer {
|
|
name: "data"
|
|
type: "HDF5Data"
|
|
top: "data"
|
|
top: "label"
|
|
hdf5_data_param {
|
|
source: "examples/hdf5_classification/data/test.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"
|
|
}
|