Hi,
I have an image that is composed of three concatenated images. I want to operate on each image separately. To achieve this, I split an input blob to multiple output blobs using a Slice layer. But during the process the python code consumes more and more RAM memory without any GPU consumption and then crashes.
I'm using Digit and I have no idea why there is this problem.
Thanks in advance,
Jérôme
My custom network like this :
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
data_param {
batch_size: 100
}
}
layer {
name: "slice_pair"
type: "Slice"
bottom: "data"
top: "data_r"
top: "data_r1"
top: "data_r2"
slice_param {
axis:1
slice_point:1
slice_point:2
}
}
layer {
name: "conv1"
type: "Convolution"
bottom: "data_r"
top: "conv1"
....
}
...