Crash on ArgMax layer time by time

24 views
Skip to first unread message

korabe...@smedx.com

unread,
Apr 7, 2016, 2:57:02 AM4/7/16
to Caffe Users
Caffe crashes on forward phase on ArgMax layer
sometimes on 1st batch, sometimes on 2nd sometime not crash

my deploy:
name: "Tumors 64x64"
input: "data"
input_dim: 1024 #batch_size
input_dim: 1  #number of channels
input_dim: 64 # h-size
input_dim: 64 # w-size

...

layer {
  name: "softmax"
  type: "Softmax"
  bottom: "fc8"
  top: "softmax"
}

layer {
  name: "argmax"
  type: "ArgMax"
  bottom: "softmax"
  top: "argmax"
}

Jan

unread,
Apr 15, 2016, 10:09:58 AM4/15/16
to Caffe Users
What is the error message?

Jan
Reply all
Reply to author
Forward
0 new messages