Hello,
I found that following part of the network does not work well, until I insert fake pooling layer after reshape.
layer {
name: "conv_a"
type: "Convolution"
bottom: "data"
top: "conv_a"
convolution_param {
num_output: 256
bias_term: true
kernel_size: 1
stride: 1
}
}
layer {
name: "relu_a"
type: "ReLU"
bottom: "conv_a"
top: "relu_a"
}
layer {
name: "conv_b"
type: "Convolution"
bottom: "data"
top: "conv_b"
convolution_param {
num_output: 64
bias_term: true
kernel_size: 1
stride: 1
}
}
layer {
name: "relu_b"
type: "ReLU"
bottom: "conv_b"
top: "relu_b"
}
layer {
name: "reshape"
type: "Reshape"
bottom: "relu_b"
top: "reshape"
reshape_param {
shape {
dim: -1
dim: 256
dim: 15
dim: 15
}
}
}
layer {
name: "pool_fake"
type: "Pooling"
bottom: "reshape"
top: "pool_fake"
pooling_param {
pool: MAX
kernel_size: 1
stride: 1
}
}
layer {
name: "concat_a_b"
type: "Concat"
bottom: "pool_fake"
bottom: "relu_b"
top: "concat_a_b"
}
thanks
Jan
