Closed DVEfremov closed 7 years ago
Can you explain what this does?
vector<int> scale_dims = bottom[0]->shape();
on L28 should be <int_tp>
Correct! Thanks a lot, will fix. I'v just taken from https://github.com/BVLC/caffe/blob/master/src/caffe/layers/softmax_layer.cpp as is now I see should be int_tp
Can you put the vector<int_tp> scale_dims = bottom[0]->shape();
statement just before #ifdef ... #else ... #endif
? It seems to be common between the two branches.