the caffe train_test.prototxt
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
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.
thanks ,I'll go and have a try.