Size mismatch in SigmoidCrossEntropyLoss about voc-classification HOT 6 OPEN

It seems like if I set batch size to some number rather than 10, this problem will happen.

from voc-classification.

@jason718 I just realized that contrary to Tensorflow the batch size is hardcoded in caffe, so we need to set the batch size equal to the one specified in whatever prototxt we are providing.

For bvlc_reference_caffenet that is indeed 10. For Rotnet it's 4 etc.

from voc-classification.

@polo5 does the batch size matters in terms of final results?

from voc-classification.

To be honnest I haven't been able to get any reasonable mAP score yet so I can't say. I get a weird error (see #7 ) which when fixed gives me a 3% mAP. I've tried the models: bvlc_reference_caffenet, alexnet, rotnet. Any luck on your side?

from voc-classification.

I used this long time ago, and I have published a paper: https://jason718.github.io/project/cvpr18/main.html

from voc-classification.

It's a mismatch between batchsize( set by -bs ) and the first dimension in input layer (set in yournetwork.prototxt).

from voc-classification.