RangeLoss-Caffe icon indicating copy to clipboard operation
RangeLoss-Caffe copied to clipboard

the caffe train_test.prototxt

Open zhanglaplace opened this issue 8 years ago • 2 comments

Hi, can yo share the train_test.prototxt and solver.prototxt ? i test the center_loss on the mnist datasets but it seems the result is not well

zhanglaplace avatar Aug 24 '17 09:08 zhanglaplace

you can add this layer in your prototxt, training with centerloss and softmaxloss is ok layer { name: "range_loss" type: "RangeLoss" bottom: "ip1" bottom: "label" top: "range_loss" range_loss_param { choose_k: 2 inter_weight: 5e-4 intra_weight: 5e-5 margin: 2000 } loss_weight: 1 }

you should change "inter_weight", "intra_weight", "margin" and "loss_weight" for your training, my experiments are not enough, I'm sorry I can't help how to choose the parameters.

Charrin avatar Aug 24 '17 09:08 Charrin

thanks ,I'll go and have a try.

zhanglaplace avatar Aug 24 '17 09:08 zhanglaplace