Slice layer and infinite loop

46 views
Skip to first unread message

Jerome Pasquet

unread,
Oct 30, 2015, 11:14:59 AM10/30/15
to Caffe Users
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"
  ....
}
...
Reply all
Reply to author
Forward
0 new messages