mirror of
https://github.com/zhunzhong07/person-re-ranking.git
synced 2025-06-03 14:59:19 +08:00
99 lines
1.4 KiB
Plaintext
99 lines
1.4 KiB
Plaintext
name: "LogisticRegressionNet"
|
|
layer {
|
|
name: "data"
|
|
type: "HDF5Data"
|
|
top: "data"
|
|
top: "label"
|
|
include {
|
|
phase: TRAIN
|
|
}
|
|
hdf5_data_param {
|
|
source: "examples/hdf5_classification/data/train.txt"
|
|
batch_size: 10
|
|
}
|
|
}
|
|
layer {
|
|
name: "data"
|
|
type: "HDF5Data"
|
|
top: "data"
|
|
top: "label"
|
|
include {
|
|
phase: TEST
|
|
}
|
|
hdf5_data_param {
|
|
source: "examples/hdf5_classification/data/test.txt"
|
|
batch_size: 10
|
|
}
|
|
}
|
|
layer {
|
|
name: "fc1"
|
|
type: "InnerProduct"
|
|
bottom: "data"
|
|
top: "fc1"
|
|
param {
|
|
lr_mult: 1
|
|
decay_mult: 1
|
|
}
|
|
param {
|
|
lr_mult: 2
|
|
decay_mult: 0
|
|
}
|
|
inner_product_param {
|
|
num_output: 40
|
|
weight_filler {
|
|
type: "xavier"
|
|
}
|
|
bias_filler {
|
|
type: "constant"
|
|
value: 0
|
|
}
|
|
}
|
|
}
|
|
layer {
|
|
name: "relu1"
|
|
type: "ReLU"
|
|
bottom: "fc1"
|
|
top: "fc1"
|
|
}
|
|
layer {
|
|
name: "fc2"
|
|
type: "InnerProduct"
|
|
bottom: "fc1"
|
|
top: "fc2"
|
|
param {
|
|
lr_mult: 1
|
|
decay_mult: 1
|
|
}
|
|
param {
|
|
lr_mult: 2
|
|
decay_mult: 0
|
|
}
|
|
inner_product_param {
|
|
num_output: 2
|
|
weight_filler {
|
|
type: "xavier"
|
|
}
|
|
bias_filler {
|
|
type: "constant"
|
|
value: 0
|
|
}
|
|
}
|
|
}
|
|
layer {
|
|
name: "loss"
|
|
type: "SoftmaxWithLoss"
|
|
bottom: "fc2"
|
|
bottom: "label"
|
|
top: "loss"
|
|
}
|
|
layer {
|
|
name: "accuracy"
|
|
type: "Accuracy"
|
|
bottom: "fc2"
|
|
bottom: "label"
|
|
top: "accuracy"
|
|
include {
|
|
phase: TEST
|
|
}
|
|
}
|