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"
}